Merge pull request #1964 from SciPhi-AI/feature/allow-backwards-compat

Feature/allow backwards compat
SciPhi-AI · Feb 11, 2025 · af648a5 · af648a5
2 parents dc6f0d4 + 22adf3c
commit af648a5
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 14 deletions.
diff --git a/py/core/main/services/retrieval_service.py b/py/core/main/services/retrieval_service.py
@@ -600,6 +600,7 @@ async def rag(
                 search_results=aggregated_results,
                 citations=mapped_citations,
                 metadata=metadata,
+                completion=re_labeled_text,
             )
             return rag_response
 

diff --git a/py/shared/api/models/retrieval/responses.py b/py/shared/api/models/retrieval/responses.py
@@ -107,9 +107,9 @@ class Config:
 
 
 class RAGResponse(BaseModel):
-    generated_answer: str = Field(
-        ..., description="The generated completion from the RAG process"
-    )
+    # generated_answer: str = Field(
+    #     ..., description="The generated completion from the RAG process"
+    # )
     search_results: AggregateSearchResult = Field(
         ..., description="The search results used for the RAG process"
     )
@@ -121,17 +121,11 @@ class RAGResponse(BaseModel):
         default_factory=dict,
         description="Additional data returned by the LLM provider",
     )
-
-    # Provide a deprecated alias property.
-    @property
-    @deprecated(version="3.4.1", reason="Use 'generated_answer' instead")
-    def completion(self) -> str:
-        return self.generated_answer
-
-    @completion.setter
-    @deprecated(version="3.4.1", reason="Use 'generated_answer' instead")
-    def completion(self, value: str):
-        self.generated_answer = value
+    completion: str = Field(
+        ...,
+        description="The generated completion from the RAG process",
+        deprecated=True,
+    )
 
     class Config:
         json_schema_extra = {