Updated to latest API release.

Author: kirk-marple

documents/agent/GetAgent.graphql

@@ -230,6 +230,7 @@ query GetAgent($id: ID!, $correlationId: String) {
     timeout
     prompt
     scratchpad
+    researchDepth
     callbackUri
   }
 }

documents/agent/QueryAgents.graphql

@@ -229,6 +229,7 @@ query QueryAgents($filter: AgentFilter, $correlationId: String) {
       timeout
       prompt
       scratchpad
+      researchDepth
       callbackUri
     }
   }

graphlit_api/__init__.py

@@ -699,6 +699,7 @@
 from .enrich_products import EnrichProducts, EnrichProductsEnrichProducts
 from .enums import (
     AgentChannelTypes,
+    AgentResearchDepths,
     AgentTypes,
     AlertTypes,
     AnthropicEffortLevels,
@@ -5461,6 +5462,7 @@
     "AgentChannelTypes",
     "AgentFilter",
     "AgentInput",
+    "AgentResearchDepths",
     "AgentSchedulePolicyInput",
     "AgentTriggerFilterInput",
     "AgentTypes",

graphlit_api/enums.py

@@ -388,6 +388,13 @@ class TwelveLabsSegmentationMethods(str, Enum):
     FIXED = "FIXED"
 
 
+class AgentResearchDepths(str, Enum):
+    QUICK = "QUICK"
+    STANDARD = "STANDARD"
+    DEEP = "DEEP"
+    EXHAUSTIVE = "EXHAUSTIVE"
+
+
 class TimedPolicyRecurrenceTypes(str, Enum):
     ONCE = "ONCE"
     REPEAT = "REPEAT"

graphlit_api/get_agent.py

@@ -8,6 +8,7 @@
 from .base_model import BaseModel
 from .enums import (
     AgentChannelTypes,
+    AgentResearchDepths,
     AgentTypes,
     ContentTypes,
     EntityState,
@@ -47,6 +48,7 @@ class GetAgentAgent(BaseModel):
     timeout: Optional[Any]
     prompt: Optional[str]
     scratchpad: Optional[str]
+    research_depth: Optional[AgentResearchDepths] = Field(alias="researchDepth")
     callback_uri: Optional[Any] = Field(alias="callbackUri")
 
 

graphlit_api/input_types.py

@@ -8,6 +8,7 @@
 from .base_model import BaseModel
 from .enums import (
     AgentChannelTypes,
+    AgentResearchDepths,
     AgentTypes,
     AlertTypes,
     AnthropicEffortLevels,
@@ -2907,6 +2908,9 @@ class AgentInput(BaseModel):
     timeout: Optional[Any] = None
     prompt: Optional[str] = None
     scratchpad: Optional[str] = None
+    research_depth: Optional[AgentResearchDepths] = Field(
+        alias="researchDepth", default=None
+    )
     channels: Optional[list["AgentChannelInput"]] = None
     connectors: Optional[list["EntityReferenceInput"]] = None
     callback_uri: Optional[Any] = Field(alias="callbackUri", default=None)
@@ -7406,6 +7410,9 @@ class AgentUpdateInput(BaseModel):
     timeout: Optional[Any] = None
     prompt: Optional[str] = None
     scratchpad: Optional[str] = None
+    research_depth: Optional[AgentResearchDepths] = Field(
+        alias="researchDepth", default=None
+    )
     channels: Optional[list["AgentChannelInput"]] = None
     connectors: Optional[list["EntityReferenceInput"]] = None
     callback_uri: Optional[Any] = Field(alias="callbackUri", default=None)

graphlit_api/operations.py

@@ -794,6 +794,7 @@
     timeout
     prompt
     scratchpad
+    researchDepth
     callbackUri
   }
 }
@@ -1031,6 +1032,7 @@
       timeout
       prompt
       scratchpad
+      researchDepth
       callbackUri
     }
   }

graphlit_api/query_agents.py

@@ -8,6 +8,7 @@
 from .base_model import BaseModel
 from .enums import (
     AgentChannelTypes,
+    AgentResearchDepths,
     AgentTypes,
     ContentTypes,
     EntityState,
@@ -51,6 +52,7 @@ class QueryAgentsAgentsResults(BaseModel):
     timeout: Optional[Any]
     prompt: Optional[str]
     scratchpad: Optional[str]
+    research_depth: Optional[AgentResearchDepths] = Field(alias="researchDepth")
     callback_uri: Optional[Any] = Field(alias="callbackUri")