ResultChat class was modified

SermetPekin · Aug 21, 2024 · 9421b69 · 9421b69
1 parent 525baf8
commit 9421b69
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/evdschat/core/chat.py b/evdschat/core/chat.py
@@ -16,7 +16,7 @@
 from evdschat.model.chatters import ModelAbstract, OpenAI
 from ..common.github_actions import PytestTesting
 from typing import TypeVar, Tuple, Union
-from .result import ResultChat 
+from .result import ResultChat  , Status 
 
 # Result = TypeVar("Result")
 Notes = TypeVar("Notes")

diff --git a/evdschat/model/chatters.py b/evdschat/model/chatters.py
@@ -29,12 +29,10 @@
 
 import pandas as pd
 
-from evdschat.core.result import Result, ResultChat , create_result 
+from evdschat.core.result import Result, ResultChat, create_result
 from evdschat.core.result import Status
 
 
-
-
 def get_myapi_url():
     from dotenv import load_dotenv
 
@@ -108,12 +106,12 @@ def eval(self, kw: dict, permitted=None) -> Union[Tuple[Any, str], None]:
             del kw["notes"]
         try:
             result = self.retrieve_fnc(kw["index"], **nkw)
-            res = create_result(result)
+            res = create_result(result, status=Status.success)
             return res, notes
         except Exception:
             traceback.print_exc()
 
-        return create_result(None, "Eval failed") , str("")
+        return create_result(None, status=Status.failed, reason="Eval failed"), str("")
 
     def decide_caller(self):
         if self.test: