- RAG & SPIN Specialist improvements

2025-04-22 13:49:38 +02:00
parent 4bf12db142
commit 9652d0bff9
12 changed files with 24 additions and 41 deletions
--- a/eveai_chat_workers/outputs/rag/rag_v1_0.py
+++ b/eveai_chat_workers/outputs/rag/rag_v1_0.py
@@ -4,6 +4,6 @@ from pydantic import BaseModel, Field


 class RAGOutput(BaseModel):
-    answer: Optional[str] = Field(None, description="Answer to the questions asked")
-    citations: Optional[List[str]] = Field(None, description="A list of sources used in generating the answer")
-    insufficient_info: Optional[bool] = Field(None, description="An indication if there's insufficient information to answer")
+    answer: str = Field(None, description="Answer to the questions asked")
+    insufficient_info: bool = Field(None, description="An indication if there's insufficient information to answer")
+