Fix for teacher_runtime

HumanSignal · Mar 26, 2024 · c240b48 · c240b48
1 parent e631d0f
commit c240b48
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 1 deletion.
diff --git 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
@@ -315,7 +315,6 @@ def improve(
 Summarize your analysis about incorrect predictions and suggest changes to the prompt.""",
             }
         ]
-
         reasoning = runtime.execute(messages)
 
         messages += [