.

arthurbrenno · Feb 8, 2025 · d928701 · d928701
1 parent a36e8f1
commit d928701
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/src/intellibricks/agents/agents.py b/src/intellibricks/agents/agents.py
@@ -580,7 +580,7 @@ async def _run_logic_async(
                 stop_sequences=self.generation_config.stop_sequences,
                 cache_config=self.generation_config.cache_config,
                 trace_params=trace_params or self.generation_config.trace_params,
-                general_web_search=self.generation_config.general_web_search,
+                grounding=self.generation_config.grounding,
                 language=self.generation_config.language,
                 timeout=self.generation_config.timeout,
             )
@@ -603,7 +603,7 @@ async def _run_logic_async(
             stop_sequences=self.generation_config.stop_sequences,
             cache_config=self.generation_config.cache_config,
             trace_params=trace_params or self.generation_config.trace_params,
-            general_web_search=self.generation_config.general_web_search,
+            grounding=self.generation_config.grounding,
             language=self.generation_config.language,
             timeout=self.generation_config.timeout,
         )

diff --git a/src/intellibricks/llms/types.py b/src/intellibricks/llms/types.py
@@ -198,7 +198,7 @@
     "google/genai/gemini-2.0-flash-exp",
     "google/genai/gemini-2.0-flash",
     "google/genai/gemini-2.0-flash-lite-preview-02-05",
-    "google/genai/gemini-2.0-flash-thinking-exp-01-21"
+    "google/genai/gemini-2.0-flash-thinking-exp-01-21",
 ]
 
 VertexAIModelType: TypeAlias = Literal[
@@ -215,7 +215,7 @@
     "google/vertexai/gemini-pro-experimental",
     "google/vertexai/gemini-2.0-flash",
     "google/vertexai/gemini-2.0-flash-lite-preview-02-05",
-    "google/vertexai/gemini-2.0-flash-thinking-exp-01-21"
+    "google/vertexai/gemini-2.0-flash-thinking-exp-01-21",
 ]
 
 GoogleModelType: TypeAlias = Literal[
@@ -396,7 +396,7 @@ class GenerationConfig(msgspec.Struct, frozen=True, kw_only=True):
         cache_config: Configuration for response caching.
         trace_params: Parameters for request tracing and analytics.
         tools: List of functions available for tool calling.
-        general_web_search: Whether to enable web search augmentation.
+        grounding: Whether to enable web search augmentation.
         language: Output language for localization. Defaults to English.
         timeout: Maximum time in seconds to wait for completion.
 
@@ -516,7 +516,7 @@ class GenerationConfig(msgspec.Struct, frozen=True, kw_only=True):
         ),
     ] = msgspec.field(default=None)
 
-    general_web_search: Annotated[
+    grounding: Annotated[
         Optional[bool],
         msgspec.Meta(
             title="General web search",