alicezhai commited on
Commit
7a53e27
1 Parent(s): 2781bea

Update modules/models/ChuanhuAgent.py (#7)

Browse files

- Update modules/models/ChuanhuAgent.py (530235b233afe71ab508a8f7795d6de84da6a104)

Files changed (1) hide show
  1. modules/models/ChuanhuAgent.py +1 -1
modules/models/ChuanhuAgent.py CHANGED
@@ -57,7 +57,7 @@ class ChuanhuAgent_Client(BaseLLMModel):
57
  self.text_splitter = TokenTextSplitter(chunk_size=500, chunk_overlap=30)
58
  self.api_key = openai_api_key
59
  self.llm = ChatOpenAI(openai_api_key=openai_api_key, temperature=0, model_name=default_chuanhu_assistant_model, openai_api_base=os.environ.get("OPENAI_API_BASE", None))
60
- self.cheap_llm = ChatOpenAI(openai_api_key=openai_api_key, temperature=0, model_name="gpt-3.5-turbo", openai_api_base=os.environ.get("OPENAI_API_BASE", None))
61
  PROMPT = PromptTemplate(template=SUMMARIZE_PROMPT, input_variables=["text"])
62
  self.summarize_chain = load_summarize_chain(self.cheap_llm, chain_type="map_reduce", return_intermediate_steps=True, map_prompt=PROMPT, combine_prompt=PROMPT)
63
  self.index_summary = None
 
57
  self.text_splitter = TokenTextSplitter(chunk_size=500, chunk_overlap=30)
58
  self.api_key = openai_api_key
59
  self.llm = ChatOpenAI(openai_api_key=openai_api_key, temperature=0, model_name=default_chuanhu_assistant_model, openai_api_base=os.environ.get("OPENAI_API_BASE", None))
60
+ self.cheap_llm = ChatOpenAI(openai_api_key=openai_api_key, temperature=0, model_name="gpt-4o", openai_api_base=os.environ.get("OPENAI_API_BASE", None))
61
  PROMPT = PromptTemplate(template=SUMMARIZE_PROMPT, input_variables=["text"])
62
  self.summarize_chain = load_summarize_chain(self.cheap_llm, chain_type="map_reduce", return_intermediate_steps=True, map_prompt=PROMPT, combine_prompt=PROMPT)
63
  self.index_summary = None