diff --git a/chain/gpt_memory.py b/chain/gpt_memory.py index 12835f0..9ee30c7 100644 --- a/chain/gpt_memory.py +++ b/chain/gpt_memory.py @@ -38,7 +38,7 @@ def redis_llm_chain_factory(self): input_variables=["chat_history", "human_input"], template=self.template) llm_chain = LLMChain( llm=OpenAI(model_name="gpt-3.5-turbo", openai_api_key=self.openai_api_key, streaming=True, - callbacks=[StreamingStdOutCallbackHandler()]), + callbacks=[StreamingStdOutCallbackHandler()],openai_api_base=self.openai_base_url), prompt=prompt, verbose=True, memory=memory,