alihan/browser-use-web-ui
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

chore: rename 'gemini' to 'google' for consistency

Browse Source
This commit is contained in:
marginal23326
2025-01-31 15:33:40 +06:00
parent 037f8e5cc0
commit 0dfecfabac
4 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_browser_use.py
View File

@@ -127,7 +127,7 @@ async def test_browser_use_custom():
)
# llm = utils.get_llm_model(
# provider="gemini",
# provider="google",
# model_name="gemini-2.0-flash-exp",
# temperature=1.0,
# api_key=os.getenv("GOOGLE_API_KEY", "")

8
tests/test_llm_api.py
View File

@@ -37,7 +37,7 @@ def get_env_value(key, provider):
env_mappings = {
"openai": {"api_key": "OPENAI_API_KEY", "base_url": "OPENAI_ENDPOINT"},
"azure_openai": {"api_key": "AZURE_OPENAI_API_KEY", "base_url": "AZURE_OPENAI_ENDPOINT"},
"gemini": {"api_key": "GOOGLE_API_KEY"},
"google": {"api_key": "GOOGLE_API_KEY"},
"deepseek": {"api_key": "DEEPSEEK_API_KEY", "base_url": "DEEPSEEK_ENDPOINT"},
"mistral": {"api_key": "MISTRAL_API_KEY", "base_url": "MISTRAL_ENDPOINT"},
}
@@ -92,9 +92,9 @@ def test_openai_model():
config = LLMConfig(provider="openai", model_name="gpt-4o")
test_llm(config, "Describe this image", "assets/examples/test.png")
def test_gemini_model():
def test_google_model():
# Enable your API key first if you haven't: https://ai.google.dev/palm_docs/oauth_quickstart
config = LLMConfig(provider="gemini", model_name="gemini-2.0-flash-exp")
config = LLMConfig(provider="google", model_name="gemini-2.0-flash-exp")
test_llm(config, "Describe this image", "assets/examples/test.png")
def test_azure_openai_model():
@@ -123,7 +123,7 @@ def test_mistral_model():
if __name__ == "__main__":
# test_openai_model()
# test_gemini_model()
# test_google_model()
# test_azure_openai_model()
#test_deepseek_model()
# test_ollama_model()