Skip to content

Commit b68a975

Browse files
committed
Add conversation to helper functions
1 parent ced2304 commit b68a975

File tree

2 files changed

+32
-11
lines changed

2 files changed

+32
-11
lines changed

agixtsdk/__init__.py

Lines changed: 31 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -245,38 +245,59 @@ def prompt_agent(
245245
except Exception as e:
246246
return self.handle_error(e)
247247

248-
def instruct(self, agent_name: str, prompt: str) -> str:
248+
def instruct(self, agent_name: str, user_input: str, conversation: str) -> str:
249249
return self.prompt_agent(
250250
agent_name=agent_name,
251251
prompt_name="instruct",
252-
prompt_args={"user_input": prompt},
252+
prompt_args={
253+
"user_input": user_input,
254+
"disable_memory": True,
255+
"conversation_name": conversation,
256+
},
253257
)
254258

255-
def chat(self, agent_name: str, prompt: str) -> str:
259+
def chat(
260+
self,
261+
agent_name: str,
262+
user_input: str,
263+
conversation: str,
264+
context_results: int = 4,
265+
) -> str:
256266
return self.prompt_agent(
257267
agent_name=agent_name,
258268
prompt_name="Chat",
259-
prompt_args={"user_input": prompt, "context_reslts": 4},
269+
prompt_args={
270+
"user_input": user_input,
271+
"context_results": context_results,
272+
"conversation_name": conversation,
273+
"disable_memory": True,
274+
},
260275
)
261276

262-
def smartinstruct(self, agent_name: str, prompt: str) -> str:
277+
def smartinstruct(self, agent_name: str, user_input: str, conversation: str) -> str:
263278
return self.run_chain(
264279
chain_name="Smart Instruct",
265-
user_input=prompt,
280+
user_input=user_input,
266281
agent_name=agent_name,
267282
all_responses=False,
268283
from_step=1,
269-
chain_args={},
284+
chain_args={
285+
"conversation_name": conversation,
286+
"disable_memory": True,
287+
},
270288
)
271289

272-
def smartchat(self, agent_name: str, prompt: str) -> str:
290+
def smartchat(self, agent_name: str, user_input: str, conversation: str) -> str:
273291
return self.run_chain(
274292
chain_name="Smart Chat",
275-
user_input=prompt,
293+
user_input=user_input,
276294
agent_name=agent_name,
277295
all_responses=False,
278296
from_step=1,
279-
chain_args={},
297+
chain_args={
298+
"conversation_name": conversation,
299+
"disable_memory": True,
300+
},
280301
)
281302

282303
def get_commands(self, agent_name: str) -> Dict[str, Dict[str, bool]]:

setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88

99
setup(
1010
name="agixtsdk",
11-
version="0.0.18",
11+
version="0.0.19",
1212
description="The AGiXT SDK for Python.",
1313
long_description=long_description,
1414
long_description_content_type="text/markdown",

0 commit comments

Comments
 (0)