fix response body

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

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/lib/model/fasttext.py b/lib/model/fasttext.py
@@ -42,5 +42,5 @@ def respond(self, docs: Union[List[schemas.Message], schemas.Message]) -> List[s
             detected_langs.append({'language': model_language, 'script': model_script, 'score': model_certainty})  
 
         for doc, detected_lang in zip(docs, detected_langs):
-            doc.response = detected_lang
+            doc.body.hash_value = detected_lang
         return docs
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.hash_value = vector
+            doc.body.hash_value = vector
         return docs
 
     def vectorize(self, texts: List[str]) -> List[List[float]]: