GenDoc / src /tools /llms.py
YvesP's picture
initial commit
4cf88e8
from langchain.llms import OpenAI
import os
OpenAI_KEY = "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"
os.environ["OPENAI_API_KEY"] = OpenAI_KEY
openai_llm = OpenAI(temperature=0)
SERPAPI_API_KEY = "dba90c4ecfa942f37e2b9eb2e7c6600ef7fb5c02ab8bbfacef426773df14c06b"
os.environ["SERPAPI_API_KEY"] = SERPAPI_API_KEY
"""
HF_API_KEY = "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"
hf_llm = HuggingFaceHub(repo_id="google/flan-t5-small",
model_kwargs={"temperature": 0, "max_length": 1000},
huggingfacehub_api_token=HF_API_KEY)
"""