diff --git a/app/core/stylist_agent_server.py b/app/core/stylist_agent_server.py index 0c9323c..e373785 100644 --- a/app/core/stylist_agent_server.py +++ b/app/core/stylist_agent_server.py @@ -258,7 +258,7 @@ class AsyncStylistAgent: context += "\nPlease recommend the next single item based on the selected items, user's request, and style guide." return context - async def run_styling_process(self, request_summary, stylist_path, start_outfit=None, user_id="test"): + async def run_styling_process(self, request_summary, stylist_path, start_outfit=None, user_id="test", callback_url=""): if start_outfit is None: start_outfit = [] self.outfit_items = start_outfit if start_outfit else [] @@ -365,7 +365,7 @@ class AsyncStylistAgent: 'Connection': "keep-alive", 'Content-Type': "application/json" } - url = 'https://83aa2db8e006.ngrok-free.app/api/style/callback' + url = f'{callback_url}/api/style/callback' response = post_request(url=url, data=json.dumps(response_data), headers=headers) logger.info(response.text) return response_data diff --git a/app/server/ChatbotAgent/agent_server.py b/app/server/ChatbotAgent/agent_server.py index 8dd9ee7..1605cd4 100644 --- a/app/server/ChatbotAgent/agent_server.py +++ b/app/server/ChatbotAgent/agent_server.py @@ -22,6 +22,7 @@ class AgentRequestModel(BaseModel): user_id: str num_outfits: int stylist_path: str + callback_url: str class LCAgent(ls.LitAPI): @@ -79,7 +80,8 @@ class LCAgent(ls.LitAPI): stylist_name=request.stylist_path, start_outfit=[], num_outfits=request.num_outfits, - user_id=request.user_id) + user_id=request.user_id, + callback_url=request.callback_url) logger.info("\n--- Final Recommendation Results ---") for i, path in enumerate(recommendation_results.get("successful_outfits", [])): @@ -101,7 +103,7 @@ class LCAgent(ls.LitAPI): return summary async def recommend_outfit(self, request_summary: str, stylist_name: str, start_outfit=None, num_outfits: int = 1, - user_id: str = "test"): + user_id: str = "test", callback_url: str = None): """ 基于用户的对话历史和需求,推荐一套搭配。 @@ -119,7 +121,8 @@ class LCAgent(ls.LitAPI): request_summary=request_summary, stylist_path=stylist_name, start_outfit=start_outfit, - user_id=user_id + user_id=user_id, + callback_url=callback_url ) tasks.append(task) print(f"--- Starting {num_outfits} concurrent outfit generation tasks. ---") diff --git a/app/server/utils/request_post.py b/app/server/utils/request_post.py index 7af3a90..1679588 100644 --- a/app/server/utils/request_post.py +++ b/app/server/utils/request_post.py @@ -33,7 +33,7 @@ def post_request(url, data=None, json_data=None, headers=None, auth=None, timeou if __name__ == '__main__': - url = 'https://83aa2db8e006.ngrok-free.app/api/style/callback' + url = 'https://0dd6f6504aff.ngrok-free.app/api/style/callback' object_data = { 'outfit_id': "test",