Rebase main

jlmartinnavarro · Aug 22, 2024 · b7b9893 · b7b9893
1 parent 767f29c
commit b7b9893
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/backend/utils/llm.py b/backend/utils/llm.py
@@ -281,6 +281,10 @@ def num_tokens_from_string(string: str) -> int:
 # ***************************************************
 
 
+class ContextOutput(BaseModel):
+    topics: List[str] = Field(default=[], description="List of topics.")
+
+
 class RequiresContext(BaseModel):
     value: bool = Field(description="Based on the conversation, this tells if context is needed to respond")