mirror/lume
1
0
Fork 0
mirror of https://github.com/trycua/lume.git synced 2026-01-06 04:20:03 -06:00
Code Issues Packages Projects Releases Wiki Activity

add cua adapter support for gemini

Browse Source
This commit is contained in:
Sarina Li
2025-12-19 12:34:16 -05:00
parent eb89fd937f
commit 05b1e5a984
1 changed files with 16 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
libs/python/agent/agent/adapters/cua_adapter.py
View File

@@ -24,6 +24,10 @@ class CUAAdapter(CustomLLM):
if "anthropic/" in model:
model = f"anthropic/{self._normalize_model(model)}"
api_base = api_base.removesuffix("/v1")
elif "gemini/" in model or "google/" in model:
# Route to Gemini pass-through endpoint
model = f"gemini/{self._normalize_model(model)}"
api_base = api_base.removesuffix("/v1") + "/gemini"
else:
model = f"openai/{self._normalize_model(model)}"
@@ -35,6 +39,10 @@ class CUAAdapter(CustomLLM):
"stream": False,
}
# Forward tools if provided
if "tools" in kwargs:
params["tools"] = kwargs["tools"]
if "optional_params" in kwargs:
params.update(kwargs["optional_params"])
del kwargs["optional_params"]
@@ -71,6 +79,10 @@ class CUAAdapter(CustomLLM):
if "anthropic/" in model:
model = f"anthropic/{self._normalize_model(model)}"
api_base = api_base.removesuffix("/v1")
elif "gemini/" in model or "google/" in model:
# Route to Gemini pass-through endpoint
model = f"gemini/{self._normalize_model(model)}"
api_base = api_base.removesuffix("/v1") + "/gemini"
else:
model = f"openai/{self._normalize_model(model)}"
@@ -82,6 +94,10 @@ class CUAAdapter(CustomLLM):
"stream": False,
}
# Forward tools if provided
if "tools" in kwargs:
params["tools"] = kwargs["tools"]
if "optional_params" in kwargs:
params.update(kwargs["optional_params"])
del kwargs["optional_params"]