Improve GEMINI

ring.py

@@ -10,6 +10,7 @@ from openworm_ai.utils.llms import get_llm as get_llm_via_openworm_ai
 
 from openworm_ai.utils.llms import requires_openai_key
 from openworm_ai.utils.llms import get_openai_api_key 
+from openworm_ai.utils.llms import get_gemini_api_key
 
 
 from openworm_ai.utils.llms import LLM_GPT35
@@ -42,10 +43,6 @@ 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"]

test.py

@@ -2,7 +2,6 @@ from ring import LLM_GPT4o
 from ring import LLM_GPT35
 from ring import LLM_LLAMA2
 from ring import LLM_GEMINI
-from ring import LLM_AI21
 from ring import LLM_CLAUDE2
 from ring import LLM_COHERE