From f828f9991b0109cfcda1b0e590eab426647ffdef Mon Sep 17 00:00:00 2001 From: Kenneth Reitz Date: Wed, 30 Oct 2024 18:43:29 -0400 Subject: [PATCH] Refactor create_conversation function to accept additional keyword arguments for flexibility --- simplemind/__init__.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 2 deletions(-) diff --git a/simplemind/__init__.py b/simplemind/__init__.py index b3bc4e2..c58217f 100644 --- a/simplemind/__init__.py +++ b/simplemind/__init__.py @@ -55,13 +55,20 @@ class Session: def create_conversation( - llm_model=None, llm_provider=None, *, plugins: Optional[List[BasePlugin]] = None + *, + llm_model=None, + llm_provider=None, + plugins: Optional[List[BasePlugin]] = None, + **kwargs, ): """Create a new conversation.""" + # Note: kwargs are here to eat up any extra arguments passed in from sessions. + # Create the conversation. conversation = Conversation( - llm_model=llm_model, llm_provider=llm_provider or settings.DEFAULT_LLM_PROVIDER + llm_model=llm_model, + llm_provider=llm_provider or settings.DEFAULT_LLM_PROVIDER, ) # Add plugins to the conversation.