From c240b48c528d9dd6be716a0de846f0b336a01aca Mon Sep 17 00:00:00 2001
From: nik <nik@heartex.net>
Date: Tue, 26 Mar 2024 01:30:45 +0000
Subject: [PATCH] Fix for teacher_runtime

---
 adala/agents/base.py  | 1 +
 adala/skills/_base.py | 1 -
 2 files changed, 1 insertion(+), 1 deletion(-)

diff --git a/adala/agents/base.py b/adala/agents/base.py
index ac28795e..bf428a80 100644
--- a/adala/agents/base.py
+++ b/adala/agents/base.py
@@ -212,6 +212,7 @@ def get_teacher_runtime(self, runtime: Optional[str] = None) -> Runtime:
             raise ValueError(f"Teacher Runtime is requested, but it was not set."
                              f"Please provide a teacher runtime in the agent's constructor explicitly:"
                              f"agent = Agent(..., teacher_runtimes={{'default': OpenAIChatRuntime(model='gpt-4')}})")
+        return runtime
 
     def run(
         self, input: InternalDataFrame = None, runtime: Optional[str] = None
diff --git a/adala/skills/_base.py b/adala/skills/_base.py
index ffefcd04..ecf29ddf 100644
--- a/adala/skills/_base.py
+++ b/adala/skills/_base.py
@@ -315,7 +315,6 @@ def improve(
 Summarize your analysis about incorrect predictions and suggest changes to the prompt.""",
             }
         ]
-
         reasoning = runtime.execute(messages)
 
         messages += [