Skip to content

Commit c60a04c

Browse files
committed
injected user id to prompt for agent reference
1 parent eabfc38 commit c60a04c

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

api/llm/agent.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@ def get_agent():
4646
return _agent_executor
4747

4848

49-
def _build_message_with_context(message: str, selected_images: Optional[List[dict]]) -> str:
49+
def _build_message_with_context(message: str, selected_images: Optional[List[dict]], user_id: str) -> str:
5050
"""Build the full message with image context if provided."""
5151
if not selected_images or len(selected_images) == 0:
5252
return message
@@ -60,7 +60,7 @@ def _build_message_with_context(message: str, selected_images: Optional[List[dic
6060
image_context += f" URL: {img.get('url')}\n"
6161
image_context += "\n"
6262

63-
return message + image_context
63+
return message + image_context + f"\n\nUser ID: {user_id}"
6464

6565

6666
def _extract_agent_response(response) -> str:
@@ -189,7 +189,7 @@ def chat_with_agent(message: str, user_id: str = "default", selected_images: Opt
189189
agent = get_agent()
190190

191191
# Prepare the message with context
192-
full_message = _build_message_with_context(message, selected_images)
192+
full_message = _build_message_with_context(message, selected_images, user_id)
193193

194194
# Configure thread ID for conversation continuity
195195
config = {"configurable": {"thread_id": user_id}}

0 commit comments

Comments
 (0)