From 494cdfa025211f0d2510d5c29023f8fe24d7c4d2 Mon Sep 17 00:00:00 2001 From: Douglas Reid Date: Tue, 29 Aug 2023 15:15:20 -0700 Subject: [PATCH] fix(agents): use proper init names for message selectors --- src/steamship/agents/examples/example_assistant.py | 2 +- src/steamship/agents/examples/example_assistant_with_caching.py | 2 +- src/steamship/agents/examples/example_react_assistant.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/steamship/agents/examples/example_assistant.py b/src/steamship/agents/examples/example_assistant.py index aebfdff63..a40f6f091 100644 --- a/src/steamship/agents/examples/example_assistant.py +++ b/src/steamship/agents/examples/example_assistant.py @@ -22,7 +22,7 @@ def __init__(self, **kwargs): DalleTool(), ], llm=ChatOpenAI(self.client, temperature=0), - conversation_memory=MessageWindowMessageSelector(k=2), + message_selector=MessageWindowMessageSelector(k=2), ) ) diff --git a/src/steamship/agents/examples/example_assistant_with_caching.py b/src/steamship/agents/examples/example_assistant_with_caching.py index dffcea24e..bd2a9c895 100644 --- a/src/steamship/agents/examples/example_assistant_with_caching.py +++ b/src/steamship/agents/examples/example_assistant_with_caching.py @@ -22,7 +22,7 @@ def __init__(self, **kwargs): DalleTool(), ], llm=ChatOpenAI(self.client, temperature=0), - conversation_memory=MessageWindowMessageSelector(k=2), + message_selector=MessageWindowMessageSelector(k=2), ) ) diff --git a/src/steamship/agents/examples/example_react_assistant.py b/src/steamship/agents/examples/example_react_assistant.py index 7bdd9d483..956ae95b6 100644 --- a/src/steamship/agents/examples/example_react_assistant.py +++ b/src/steamship/agents/examples/example_react_assistant.py @@ -24,7 +24,7 @@ def __init__(self, **kwargs): llm=OpenAI( self.client, temperature=0, model_name="gpt-4-0314" ), # MUST PIN to older model - conversation_memory=MessageWindowMessageSelector(k=2), + message_selector=MessageWindowMessageSelector(k=2), ) )