shaytan/ai-agents-masterclass
1
0
Fork 0
mirror of https://github.com/coleam00/ai-agents-masterclass.git synced 2025-11-29 00:23:14 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #35 from jeffwiegner/patch-chat_id

Browse Source 
Getting chat_id to use as session_id
This commit is contained in:
Cole Medin
2025-01-14 10:15:41 -06:00
committed by GitHub
parent f1c4ef0911 a91332afd9
commit 1879176f4a
1 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
local-ai-packaged/n8n_pipe.py
View File

@@ -13,6 +13,15 @@ import os
import time
import requests
def extract_event_info(event_emitter) -> tuple[Optional[str], Optional[str]]:
if not event_emitter or not event_emitter.__closure__:
return None, None
for cell in event_emitter.__closure__:
if isinstance(request_info := cell.cell_contents, dict):
chat_id = request_info.get("chat_id")
message_id = request_info.get("message_id")
return chat_id, message_id
return None, None
class Pipe:
class Valves(BaseModel):
@@ -75,7 +84,7 @@ class Pipe:
await self.emit_status(
__event_emitter__, "info", "/Calling N8N Workflow...", False
)
chat_id, _ = extract_event_info(__event_emitter__)
messages = body.get("messages", [])
# Verify a message is available
@@ -87,7 +96,7 @@ class Pipe:
"Authorization": f"Bearer {self.valves.n8n_bearer_token}",
"Content-Type": "application/json",
}
payload = {"sessionId": f"{__user__['id']} - {messages[0]['content'].split('Prompt: ')[-1][:100]}"}
payload = {"sessionId": f"{chat_id}"}
payload[self.valves.input_field] = question
response = requests.post(
self.valves.n8n_url, json=payload, headers=headers