Adds ChatGoogleGenerativeAI

.gitignore

@@ -4,3 +4,4 @@
 /models/LEMS_Sim_IClamp_GenericMuscleCell.xml
 /report.Sim_IClamp_GenericMuscleCell.txt
 /report.Sim_IClamp_GenericNeuronCell.txt
+*v.dat

app.py

@@ -7,6 +7,7 @@ from ring import LLM_GPT35
 from ring import LLM_GPT4
 from ring import LLM_GPT4o
 from ring import LLM_LLAMA2
+from ring import LLM_GEMINI
 from ring import PREF_ORDER_LLMS
 
 
@@ -87,7 +88,7 @@ with tab_panel:
         elif submitted:
             response = generate_panel_response(
                 text,
-                llm_panelists=[LLM_GPT35, LLM_GPT4],
+                llm_panelists=[LLM_GPT35, LLM_GPT4, LLM_GEMINI],
                 llm_panel_chair=LLM_GPT4o,
                 temperature=temperature,
             )

requirements.txt

@@ -9,4 +9,5 @@ neuromllite
 matplotlib
 pylems
 pyneuroml
-paper-qa==4.0.0rc8
+paper-qa==4.0.0rc8
+langchain-google-genai

ring.py

@@ -10,10 +10,11 @@ LLM_GPT35 = "GPT3.5"
 LLM_GPT4 = "GPT4"
 LLM_GPT4o = "GPT4o"
 LLM_LLAMA2 = "LLAMA2"
+LLM_GEMINI = "Gemini"
 
 OPENAI_LLMS = [LLM_GPT35, LLM_GPT4, LLM_GPT4o]
 
-PREF_ORDER_LLMS = (LLM_LLAMA2, LLM_GPT35, LLM_GPT4, LLM_GPT4o)
+PREF_ORDER_LLMS = (LLM_GEMINI, LLM_LLAMA2, LLM_GPT35, LLM_GPT4, LLM_GPT4o)
 
 
 def requires_openai_key(llm_ver):
@@ -40,6 +41,13 @@ def get_llamaapi_key():
     return llamaapi_key
 
 
+def get_gemini_api_key():
+
+    gemini_api_key = os.environ.get("GEMINIAPI_KEY")
+
+    return gemini_api_key
+
+
 def get_llm(llm_ver, temperature):
 
     if llm_ver == LLM_GPT35:
@@ -78,6 +86,14 @@ def get_llm(llm_ver, temperature):
 
         llm = ChatLlamaAPI(client=llama)
 
+    elif llm_ver == LLM_GEMINI:
+
+        from langchain_google_genai import ChatGoogleGenerativeAI
+
+        llm = ChatGoogleGenerativeAI(
+            model="gemini-pro", google_api_key=get_gemini_api_key()
+        )
+
     return llm
 
 

test.py

@@ -2,19 +2,27 @@ from ring import LLM_GPT35
 from ring import LLM_GPT4
 from ring import LLM_GPT4o
 from ring import LLM_LLAMA2
+from ring import LLM_GEMINI
 from ring import PREF_ORDER_LLMS
 
-
 from ring import generate_response
 
+import sys
+
 question = "What is the most common type of neuron in the brain?"
 
 llm_ver = LLM_GPT4o
 
+if "-g" in sys.argv:
+    llm_ver = LLM_GEMINI
+
+
 print("--------------------------------------------------------")
 print("Asking question:\n   %s" % question)
 print("--------------------------------------------------------")
 
+print(" ... Connecting to: %s" % llm_ver)
+
 response = generate_response(question, llm_ver, temperature=0, only_celegans=False)