Avoid code duplication in context (re)creation

Author: PeterBowman

libraries/YarpPlugins/LlamaGPT/DeviceDriverImpl.cpp

@@ -97,18 +97,9 @@ bool LlamaGPT::open(yarp::os::Searchable & config)
         }
     }
 
-    // initialize the context
-    llama_context_params ctx_params = llama_context_default_params();
-    ctx_params.n_ctx = m_tokens;
-    ctx_params.n_batch = m_tokens;
-    ctx_params.no_perf = false;
-
-    ctx = llama_init_from_model(model, ctx_params);
-
-    if (!ctx)
+    if (!clear(false)) // initialize the context
     {
-        yCError(LLAMA) << "Failed to create context";
-        return 1;
+        return false;
     }
 
     // initialize the sampler

libraries/YarpPlugins/LlamaGPT/LlamaGPT.cpp

@@ -46,11 +46,13 @@ bool LlamaGPT::clear(bool preservePrompt, bool recreateContext)
 
     prev_len = 0;
 
-    if (recreateContext && ctx)
+    if (recreateContext)
     {
-        llama_free(ctx);
+        if (ctx)
+        {
+            llama_free(ctx);
+        }
 
-        // initialize the context
         llama_context_params ctx_params = llama_context_default_params();
         ctx_params.n_ctx = m_tokens;
         ctx_params.n_batch = m_tokens;
@@ -60,7 +62,7 @@ bool LlamaGPT::clear(bool preservePrompt, bool recreateContext)
 
         if (!ctx)
         {
-            yCError(LLAMA) << "Failed to recreate context";
+            yCError(LLAMA) << "Failed to (re)create context";
             return 1;
         }
     }