feat: upgrade ai-exchange to version 0.8.3 and fix tests (#34)

This commit is contained in:
Luke Alvoeiro
2024-09-02 20:40:02 -07:00
committed by GitHub
parent 9444ed16ea
commit 3c930e1766
3 changed files with 8 additions and 8 deletions
+3 -3
View File
@@ -1,4 +1,4 @@
from exchange import Exchange, Message
from exchange import Exchange, Message, CheckpointData
def ask_an_ai(input: str, exchange: Exchange, prompt: str = "", no_history: bool = True) -> Message:
@@ -61,9 +61,9 @@ def clear_exchange(exchange: Exchange, clear_tools: bool = False) -> Exchange:
"""
if clear_tools:
new_exchange = exchange.replace(messages=[], checkpoints=[], tools=())
new_exchange = exchange.replace(messages=[], checkpoint_data=CheckpointData(), tools=())
else:
new_exchange = exchange.replace(messages=[], checkpoints=[])
new_exchange = exchange.replace(messages=[], checkpoint_data=CheckpointData())
return new_exchange