From 99adca4b153abb51d49fa06c0a8ed23b8db72827 Mon Sep 17 00:00:00 2001
From: Dillon DuPont <ddupont@mit.edu>
Date: Wed, 15 Oct 2025 10:16:46 -0400
Subject: [PATCH] register gemini model

---
 libs/python/agent/agent/loops/__init__.py | 2 ++
 libs/python/agent/agent/loops/openai.py   | 3 +--
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/libs/python/agent/agent/loops/__init__.py b/libs/python/agent/agent/loops/__init__.py
index d9014259..bda16cf9 100644
--- a/libs/python/agent/agent/loops/__init__.py
+++ b/libs/python/agent/agent/loops/__init__.py
@@ -14,6 +14,7 @@ from . import opencua
 from . import internvl
 from . import holo
 from . import moondream3
+from . import gemini
 
 __all__ = [
     "anthropic", 
@@ -27,4 +28,5 @@ __all__ = [
     "internvl",
     "holo",
     "moondream3",
+    "gemini"
 ]
\ No newline at end of file
diff --git a/libs/python/agent/agent/loops/openai.py b/libs/python/agent/agent/loops/openai.py
index 4fa62e66..4cdf1f31 100644
--- a/libs/python/agent/agent/loops/openai.py
+++ b/libs/python/agent/agent/loops/openai.py
@@ -53,8 +53,7 @@ async def _prepare_tools_for_openai(tool_schemas: List[Dict[str, Any]]) -> Tools
     
     return openai_tools
 
-
-@register_agent(models=r".*computer-use-preview.*")
+@register_agent(models=r".*(^|/)computer-use-preview")
 class OpenAIComputerUseConfig:
     """
     OpenAI computer-use-preview agent configuration using liteLLM responses.