From 88c6f0618349d281476bdff3d1fe5b63b95b7146 Mon Sep 17 00:00:00 2001 From: bwmac Date: Wed, 8 Jan 2025 16:47:58 -0500 Subject: [PATCH] adds missing docstrings --- synapseclient/models/agent.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/synapseclient/models/agent.py b/synapseclient/models/agent.py index 7a32368dc..0e53f3a97 100644 --- a/synapseclient/models/agent.py +++ b/synapseclient/models/agent.py @@ -420,9 +420,12 @@ async def prompt( """ syn = Synapse.get_client(synapse_client=synapse_client) + # TODO: Iron this out. It's a little confusing. if session_id: if session_id not in self.sessions: await self.get_session_async(session_id=session_id, synapse_client=syn) + else: + self.current_session = session_id else: if not self.current_session: await self.start_session_async(synapse_client=syn) @@ -440,4 +443,5 @@ async def prompt( ) def get_chat_history(self) -> List[AgentPrompt]: """Gets the chat history for the current session.""" + # TODO: Is this the best way to do this? return self.sessions[self.current_session].chat_history