mirror/computer
1
0
Fork 0
mirror of https://github.com/trycua/computer.git synced 2026-01-04 12:30:08 -06:00
Code Issues Packages Projects Releases Wiki Activity

Update params dict

Browse Source
This commit is contained in:
Dillon DuPont
2025-11-18 15:31:30 -05:00
parent cb81fddd74
commit 8284f85950
1 changed files with 19 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -19,30 +19,28 @@ class CUAAdapter(CustomLLM):
return model.split("/", 1)[1] if model and model.startswith("cua/") else model
def completion(self, *args, **kwargs) -> ModelResponse:
params = dict(kwargs)
inner_model = self._normalize_model(params.get("model", ""))
params.update(
{
"model": f"openai/{inner_model}",
"api_base": self.base_url,
"api_key": self.api_key,
"stream": False,
}
)
params = {
"model": f"openai/{self._normalize_model(kwargs.get("model", ""))}",
"messages": kwargs.get("messages", []),
"api_base": self.base_url,
"api_key": self.api_key,
"stream": False,
}
return completion(**params) # type: ignore
async def acompletion(self, *args, **kwargs) -> ModelResponse:
params = dict(kwargs)
inner_model = self._normalize_model(params.get("model", ""))
params.update(
{
"model": f"openai/{inner_model}",
"api_base": self.base_url,
"api_key": self.api_key,
"stream": False,
}
)
return await acompletion(**params) # type: ignore
params = {
"model": f"openai/{self._normalize_model(kwargs.get("model", ""))}",
"messages": kwargs.get("messages", []),
"api_base": self.base_url,
"api_key": self.api_key,
"stream": False,
}
response = await acompletion(**params) # type: ignore
return response
def streaming(self, *args, **kwargs) -> Iterator[GenericStreamingChunk]:
params = dict(kwargs)