Spaces:
Running
Running
Test gemini 2.5
Browse files- requirements.txt +1 -0
- 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 |
-
|
|
|
|
|
|
|
| 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"
|