added logging and changed truth score

bettersg · Mar 9, 2024 · 0d14416 · 0d14416
1 parent c45eafe
commit 0d14416
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/base/schemas.py b/base/schemas.py
@@ -27,7 +27,7 @@ class VoteInitialisation(BaseModel):
 
 class Vote(BaseModel):
     category: FactCheckCategory
-    truthScore: Optional[Annotated[int, Field(ge=0, le=5)]] = None
+    truthScore: Optional[Annotated[int, Field(ge=1, le=5)]] = None
 
 class UnsupportedMessageTypeException(Exception):
     pass
diff --git a/implementation/agent.py b/implementation/agent.py
@@ -97,6 +97,7 @@ def process_call(tool_call):
                         run_id=run.id,
                         tool_outputs=tool_outputs
                     )
+        logging.info(f"Agent complete: {agent_complete}, thread_id: {thread.id}, run_id: {run.id}")
         if not agent_complete:
             raise TimeoutError("Agent did not complete in time")
         if not self.category: