CV2-2875 tweak response structure

meedan · Nov 3, 2023 · 8a32cdf · 8a32cdf
1 parent 17d0357
commit 8a32cdf
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/lib/model/generic_transformer.py b/lib/model/generic_transformer.py
@@ -25,7 +25,7 @@ def respond(self, docs: Union[List[schemas.Message], schemas.Message]) -> List[s
         vectorizable_texts = [e.body.text for e in docs]
         vectorized = self.vectorize(vectorizable_texts)
         for doc, vector in zip(docs, vectorized):
-            doc.response = vector
+            doc.hash_value = vector
         return docs
 
     def vectorize(self, texts: List[str]) -> List[List[float]]:

diff --git a/lib/schemas.py b/lib/schemas.py
@@ -12,6 +12,7 @@ class TextOutput(BaseModel):
     id: str
     callback_url: str
     text: str
+    hash_value: HashValue
 
 class VideoInput(BaseModel):
     id: str