pgleeson commited on
Commit
7b99083
·
1 Parent(s): c1f1c90

Test gemini 2.5

Browse files
Files changed (2) hide show
  1. requirements.txt +1 -0
  2. ring.py +4 -1
requirements.txt CHANGED
@@ -9,4 +9,5 @@ pylems
9
  pyneuroml
10
  langchain-google-genai
11
  langchain_cohere
 
12
  streamlit
 
9
  pyneuroml
10
  langchain-google-genai
11
  langchain_cohere
12
+ langchain_openai
13
  streamlit
ring.py CHANGED
@@ -12,6 +12,7 @@ from openworm_ai.utils.llms import LLM_GPT35
12
  from openworm_ai.utils.llms import LLM_GPT4
13
  from openworm_ai.utils.llms import LLM_GPT4o
14
  from openworm_ai.utils.llms import LLM_GEMINI_2F
 
15
  from openworm_ai.utils.llms import LLM_OLLAMA_LLAMA32_1B
16
  from openworm_ai.utils.llms import LLM_OLLAMA_MISTRAL
17
  from openworm_ai.utils.llms import LLM_OLLAMA_OLMO2_7B
@@ -24,7 +25,9 @@ OPENAI_LLMS = [LLM_GPT35, LLM_GPT4, LLM_GPT4o]
24
 
25
  OLLAMA_LLMS = [LLM_OLLAMA_LLAMA32_1B, LLM_OLLAMA_MISTRAL, LLM_OLLAMA_OLMO2_7B]
26
 
27
- OPENWORM_AI_LLMS = [LLM_GEMINI_2F] + OPENAI_LLMS + [LLM_COHERE] # + OLLAMA_LLMS
 
 
28
 
29
  LLM_LLAMA2 = "LLAMA2"
30
  # LLM_GEMINI = "gemini-1.5-pro"
 
12
  from openworm_ai.utils.llms import LLM_GPT4
13
  from openworm_ai.utils.llms import LLM_GPT4o
14
  from openworm_ai.utils.llms import LLM_GEMINI_2F
15
+ from openworm_ai.utils.llms import LLM_GEMINI_25F
16
  from openworm_ai.utils.llms import LLM_OLLAMA_LLAMA32_1B
17
  from openworm_ai.utils.llms import LLM_OLLAMA_MISTRAL
18
  from openworm_ai.utils.llms import LLM_OLLAMA_OLMO2_7B
 
25
 
26
  OLLAMA_LLMS = [LLM_OLLAMA_LLAMA32_1B, LLM_OLLAMA_MISTRAL, LLM_OLLAMA_OLMO2_7B]
27
 
28
+ GEMINI_LLMS = [LLM_GEMINI_2F, LLM_GEMINI_25F]
29
+
30
+ OPENWORM_AI_LLMS = GEMINI_LLMS + OPENAI_LLMS + [LLM_COHERE] # + OLLAMA_LLMS
31
 
32
  LLM_LLAMA2 = "LLAMA2"
33
  # LLM_GEMINI = "gemini-1.5-pro"