ilj commited on
Commit
dd0cd45
1 Parent(s): bbb957e

add gemini

Browse files
Files changed (2) hide show
  1. langchain_pipeline.py +1 -1
  2. requirements.txt +2 -0
langchain_pipeline.py CHANGED
@@ -11,7 +11,7 @@ ASTRA_DB_API_ENDPOINT = os.environ["ASTRA_DB_API_ENDPOINT"]
11
  ASTRA_DB_APPLICATION_TOKEN = os.environ["ASTRA_DB_APPLICATION_TOKEN"]
12
  OPENAI_API_KEY = os.environ["OPENAI_API_KEY"]
13
  ANTHROPIC_API_KEY = os.environ["ANTHROPIC_API_KEY"]
14
- GEMINI_API_KEY = os.environ["GEMINI_API_KEY"]
15
 
16
  collection_name = "ilj_test"
17
 
 
11
  ASTRA_DB_APPLICATION_TOKEN = os.environ["ASTRA_DB_APPLICATION_TOKEN"]
12
  OPENAI_API_KEY = os.environ["OPENAI_API_KEY"]
13
  ANTHROPIC_API_KEY = os.environ["ANTHROPIC_API_KEY"]
14
+ GOOGLE_API_KEY = os.environ["GOOGLE_API_KEY"]
15
 
16
  collection_name = "ilj_test"
17
 
requirements.txt CHANGED
@@ -6,3 +6,5 @@ langchain_astradb
6
  langchain_core
7
  langchain_openai
8
  langchain_anthropic
 
 
 
6
  langchain_core
7
  langchain_openai
8
  langchain_anthropic
9
+ langchain_google_genai
10
+ google_generativeai