YvesP commited on
Commit
ac30754
·
1 Parent(s): f178e59

Update src/tools/llms.py

Browse files
Files changed (1) hide show
  1. src/tools/llms.py +2 -12
src/tools/llms.py CHANGED
@@ -1,20 +1,10 @@
1
 
2
  from langchain.llms import OpenAI
3
 
4
- import os
5
 
6
 
7
- OpenAI_KEY = "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"
8
- os.environ["OPENAI_API_KEY"] = OpenAI_KEY
9
- openai_llm = OpenAI(temperature=0)
10
 
11
- SERPAPI_API_KEY = "dba90c4ecfa942f37e2b9eb2e7c6600ef7fb5c02ab8bbfacef426773df14c06b"
12
- os.environ["SERPAPI_API_KEY"] = SERPAPI_API_KEY
13
 
14
 
15
- """
16
- HF_API_KEY = "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"
17
- hf_llm = HuggingFaceHub(repo_id="google/flan-t5-small",
18
- model_kwargs={"temperature": 0, "max_length": 1000},
19
- huggingfacehub_api_token=HF_API_KEY)
20
- """
 
1
 
2
  from langchain.llms import OpenAI
3
 
 
4
 
5
 
6
+ openai_llm = OpenAI(temperature=0, model_name="gpt-3.5-turbo-instruct" )
7
+
 
8
 
 
 
9
 
10