Added X-Session-API-Key to nested event store (#8857)

This commit is contained in:
tofarr 2025-06-02 14:15:41 -06:00 committed by GitHub
parent aac2e51fc7
commit d03efa284a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 8 additions and 2 deletions

View File

@ -19,6 +19,7 @@ class NestedEventStore(EventStoreABC):
base_url: str
sid: str
user_id: str | None
session_api_key: str | None = None
def search_events(
self,
@ -37,7 +38,10 @@ class NestedEventStore(EventStoreABC):
search_params['limit'] = min(100, limit)
search_str = urlencode(search_params)
url = f'{self.base_url}/events{search_str}'
response = httpx.get(url)
headers = {}
if self.session_api_key:
headers['X-Session-API-Key'] = self.session_api_key
response = httpx.get(url, headers=headers)
result_set = response.json()
for result in result_set['results']:
event = event_from_dict(result)

View File

@ -124,14 +124,16 @@ class DockerNestedConversationManager(ConversationManager):
)
nested_url = self._get_nested_url(sid)
session_api_key = self._get_session_api_key_for_conversation(sid)
return AgentLoopInfo(
conversation_id=sid,
url=nested_url,
session_api_key=self._get_session_api_key_for_conversation(sid),
session_api_key=session_api_key,
event_store=NestedEventStore(
base_url=nested_url,
sid=sid,
user_id=user_id,
session_api_key=session_api_key,
),
status=ConversationStatus.STARTING
if sid in self._starting_conversation_ids