updated state and retrieval process

business_interaction_agent/utils/state.py

@@ -7,7 +7,7 @@ class State(BaseModel):
     model_config = ConfigDict(arbitrary_types_allowed=True)
 
 class StateUpdateFormatter(BaseModel):
-    business_type: dict = Field(description="The type of the business. If updated, the new one.")
+    business_type: str = Field(description="The type of the business. If updated, the new one.")
     platform: str = Field(description="The platform used for the business. If updated, the new one.")
     target_audience: str = Field(description="The target audience of the business. If updated, the new one.")
     business_goals: str = Field(description="The business goals of the business. If updated, the new one.")

business_interaction_agent/utils/utils.py

@@ -27,8 +27,8 @@ def manual_retrieval(messages, business_details):
     # === Load SentenceTransformer model ===
 
     # === Encode the query and search ===
-    query_embedding = ST.encode(str(messages)+str(business_details)).reshape(1, -1).astype('float32')
-    top_k=10
+    query_embedding = ST.encode(str(messages)).reshape(1, -1).astype('float32')
+    top_k=7
     distances, indices = index.search(query_embedding, top_k)
 
 
@@ -40,7 +40,7 @@ def manual_retrieval(messages, business_details):
             'rank': i + 1,
             'username': df.iloc[idx]['username'],
             'story': df.iloc[idx]['story'],
-            'visible_text_or_brandings': df.iloc[idx]['story'],
+            'visible_texts_or_brandings': df.iloc[idx]['visible_texts_or_brandings'],
             'likesCount': df.iloc[idx]['likesCount'],
             'commentCount': df.iloc[idx]['commentCount'],
         }
@@ -48,8 +48,8 @@ def manual_retrieval(messages, business_details):
         inner_list = []
         inner_list.append(f"[{res['rank']}]. The influencer name is: **{res['username']}** — Likes: **{res['likesCount']}**, Comments: **{res['commentCount']}**")
         inner_list.append(f"The story of that particular video is:\n{res['story']}")
-        inner_list.append(f"The branding or promotion done is:\n{res['visible_text_or_brandings']}")
+        inner_list.append(f"The branding or promotion done is:\n{res['visible_texts_or_brandings']}")
 
         outer_list.append(inner_list)
-
+    print('THE RETRIEVALS ARE:', outer_list)
     return str(outer_list).strip()