Fix gemini

ring.py

@@ -7,7 +7,7 @@ from openworm_ai.utils.llms import GENERAL_QUERY_PROMPT_TEMPLATE
 from openworm_ai.utils.llms import GENERAL_QUERY_LIMITED_PROMPT_TEMPLATE
 
 from openworm_ai.utils.llms import get_llm as get_llm_via_openworm_ai 
-from openworm_ai.utils.llms import get_gemini_api_key
+
 from openworm_ai.utils.llms import requires_openai_key
 from openworm_ai.utils.llms import get_openai_api_key 
 
@@ -24,15 +24,16 @@ from openworm_ai.utils.llms import LLM_COHERE
 
 OPENAI_LLMS = [LLM_GPT35, LLM_GPT4, LLM_GPT4o]
 
-OPENWORM_AI_LLMS = OPENAI_LLMS + [LLM_GEMINI_2F, LLM_OLLAMA_LLAMA32_1B, LLM_OLLAMA_MISTRAL, LLM_OLLAMA_OLMO2_7B, LLM_COHERE]
+OLLAMA_LLMS = [LLM_OLLAMA_LLAMA32_1B, LLM_OLLAMA_MISTRAL, LLM_OLLAMA_OLMO2_7B]
+
+OPENWORM_AI_LLMS = [LLM_GEMINI_2F] + OPENAI_LLMS + [LLM_COHERE] # + oLLAMA_LLMS
 
 LLM_LLAMA2 = "LLAMA2"
 LLM_GEMINI = "gemini-1.5-pro"
 LLM_CLAUDE2 = "Claude2.1"
 
 
-PREF_ORDER_LLMS = [LLM_GEMINI, LLM_LLAMA2] + OPENWORM_AI_LLMS + [LLM_CLAUDE2,
-    LLM_COHERE]
+PREF_ORDER_LLMS = [LLM_GEMINI] + OPENWORM_AI_LLMS + [LLM_LLAMA2, LLM_CLAUDE2]
 
 
 
@@ -41,6 +42,10 @@ 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_anthropic_key():
     anthropic_api_key = os.environ.get["ANTHROPIC_API_KEY"]