Hugging Face's logo

Spaces:
el-camino-de-santiago
/
model-pick
Sleeping

App Files Community
anmolsahai commited on
Commit
f6da619
•
1 Parent(s): e0da68f

apis

Browse files
Files changed (3) hide show
  1. __pycache__/langchain_pipeline.cpython-310.pyc +0 -0
  2. langchain_pipeline.py +1 -1
  3. requirements.txt +1 -0
__pycache__/langchain_pipeline.cpython-310.pyc CHANGED
Binary files a/__pycache__/langchain_pipeline.cpython-310.pyc and b/__pycache__/langchain_pipeline.cpython-310.pyc differ
 
langchain_pipeline.py CHANGED
@@ -19,7 +19,7 @@ embedding = OpenAIEmbeddings(model="text-embedding-ada-002")
19
 
20
  models = {
21
  "claude-3": ChatAnthropic(model='claude-3-sonnet-20240229'),
22
- "gemini-pro": ChatGoogleGenerativeAI(model="gemini-pro")
23
  }
24
 
25
  def model_names():
 
19
 
20
  models = {
21
  "claude-3": ChatAnthropic(model='claude-3-sonnet-20240229'),
22
+ "gemini-pro": ChatGoogleGenerativeAI(model="gemini-1.5-pro", temperature=0.0, google_api_key=os.getenv('GOOGLE_API_KEY'))
23
  }
24
 
25
  def model_names():
requirements.txt CHANGED
@@ -8,3 +8,4 @@ langchain_openai
8
  langchain_anthropic
9
  langchain_google_genai
10
  google_generativeai
 
 
8
  langchain_anthropic
9
  langchain_google_genai
10
  google_generativeai
11
+ langchain-google-genai==1.0.5