Updated to latest API release.

Author: kirk-marple

documents/agent/GetAgent.graphql

@@ -113,6 +113,12 @@ query GetAgent($id: ID!, $correlationId: String) {
       cron
       timeZoneId
     }
+    channels {
+      type
+      identifier
+      instructions
+      label
+    }
     timeout
     scratchpad
   }

documents/agent/QueryAgents.graphql

@@ -115,6 +115,12 @@ query QueryAgents($filter: AgentFilter, $correlationId: String) {
         cron
         timeZoneId
       }
+      channels {
+        type
+        identifier
+        instructions
+        label
+      }
       timeout
       scratchpad
     }

graphlit_api/__init__.py

@@ -664,6 +664,7 @@
 from .enrich_places import EnrichPlaces, EnrichPlacesEnrichPlaces
 from .enrich_products import EnrichProducts, EnrichProductsEnrichProducts
 from .enums import (
+    AgentChannelTypes,
     AgentTypes,
     AlertTypes,
     AnthropicEffortLevels,
@@ -956,6 +957,7 @@
 from .get_agent import (
     GetAgent,
     GetAgentAgent,
+    GetAgentAgentChannels,
     GetAgentAgentFilter,
     GetAgentAgentFilterAnd,
     GetAgentAgentFilterAndCollections,
@@ -1865,6 +1867,7 @@
 from .input_types import (
     AddressFilter,
     AddressInput,
+    AgentChannelInput,
     AgentFilter,
     AgentInput,
     AgentSchedulePolicyInput,
@@ -3043,6 +3046,7 @@
     QueryAgents,
     QueryAgentsAgents,
     QueryAgentsAgentsResults,
+    QueryAgentsAgentsResultsChannels,
     QueryAgentsAgentsResultsFilter,
     QueryAgentsAgentsResultsFilterAnd,
     QueryAgentsAgentsResultsFilterAndCollections,
@@ -5077,6 +5081,8 @@
     "AddConversationsToCollectionsAddConversationsToCollectionsContents",
     "AddressFilter",
     "AddressInput",
+    "AgentChannelInput",
+    "AgentChannelTypes",
     "AgentFilter",
     "AgentInput",
     "AgentSchedulePolicyInput",
@@ -6256,6 +6262,7 @@
     "GeometryMetadataInput",
     "GetAgent",
     "GetAgentAgent",
+    "GetAgentAgentChannels",
     "GetAgentAgentFilter",
     "GetAgentAgentFilterAnd",
     "GetAgentAgentFilterAndCollections",
@@ -7726,6 +7733,7 @@
     "QueryAgents",
     "QueryAgentsAgents",
     "QueryAgentsAgentsResults",
+    "QueryAgentsAgentsResultsChannels",
     "QueryAgentsAgentsResultsFilter",
     "QueryAgentsAgentsResultsFilterAnd",
     "QueryAgentsAgentsResultsFilterAndCollections",

graphlit_api/enums.py

@@ -1584,6 +1584,14 @@ class ModelServiceTypes(str, Enum):
     QUIVER = "QUIVER"
 
 
+class AgentChannelTypes(str, Enum):
+    VOICE = "VOICE"
+    EMAIL = "EMAIL"
+    SLACK = "SLACK"
+    TEAMS = "TEAMS"
+    DISCORD = "DISCORD"
+
+
 class AgentTypes(str, Enum):
     AGENT = "AGENT"
 
@@ -1822,6 +1830,7 @@ class GoogleModels(str, Enum):
     GEMINI_1_5_PRO_002 = "GEMINI_1_5_PRO_002"
     EMBEDDING_004 = "EMBEDDING_004"
     GEMINI_EMBEDDING_001 = "GEMINI_EMBEDDING_001"
+    GEMINI_EMBEDDING_2_PREVIEW = "GEMINI_EMBEDDING_2_PREVIEW"
     CUSTOM = "CUSTOM"
 
 

graphlit_api/get_agent.py

@@ -7,6 +7,7 @@
 
 from .base_model import BaseModel
 from .enums import (
+    AgentChannelTypes,
     AgentTypes,
     ContentTypes,
     EntityState,
@@ -35,6 +36,7 @@ class GetAgentAgent(BaseModel):
     schedule_policy: Optional["GetAgentAgentSchedulePolicy"] = Field(
         alias="schedulePolicy"
     )
+    channels: Optional[list["GetAgentAgentChannels"]]
     timeout: Optional[Any]
     scratchpad: Optional[str]
 
@@ -207,6 +209,13 @@ class GetAgentAgentSchedulePolicy(BaseModel):
     time_zone_id: Optional[str] = Field(alias="timeZoneId")
 
 
+class GetAgentAgentChannels(BaseModel):
+    type: AgentChannelTypes
+    identifier: str
+    instructions: Optional[str]
+    label: Optional[str]
+
+
 GetAgent.model_rebuild()
 GetAgentAgent.model_rebuild()
 GetAgentAgentFilter.model_rebuild()

graphlit_api/input_types.py

@@ -7,6 +7,7 @@
 
 from .base_model import BaseModel
 from .enums import (
+    AgentChannelTypes,
     AgentTypes,
     AlertTypes,
     AnthropicEffortLevels,
@@ -2776,6 +2777,7 @@ class AgentInput(BaseModel):
     )
     timeout: Optional[Any] = None
     scratchpad: Optional[str] = None
+    channels: Optional[list["AgentChannelInput"]] = None
 
 
 class CrustdataPersonDiscoveryFilterInput(BaseModel):
@@ -6776,6 +6778,13 @@ class HubSpotTasksFeedPropertiesInput(BaseModel):
     connector: Optional["EntityReferenceInput"] = None
 
 
+class AgentChannelInput(BaseModel):
+    type: AgentChannelTypes
+    identifier: str
+    instructions: Optional[str] = None
+    label: Optional[str] = None
+
+
 class MedicalStudyFacetInput(BaseModel):
     time_interval: Optional[TimeIntervalTypes] = Field(
         alias="timeInterval", default=None
@@ -6924,6 +6933,7 @@ class AgentUpdateInput(BaseModel):
     )
     timeout: Optional[Any] = None
     scratchpad: Optional[str] = None
+    channels: Optional[list["AgentChannelInput"]] = None
 
 
 class IssueFeedPropertiesInput(BaseModel):

graphlit_api/operations.py

@@ -642,6 +642,12 @@
       cron
       timeZoneId
     }
+    channels {
+      type
+      identifier
+      instructions
+      label
+    }
     timeout
     scratchpad
   }
@@ -766,6 +772,12 @@
         cron
         timeZoneId
       }
+      channels {
+        type
+        identifier
+        instructions
+        label
+      }
       timeout
       scratchpad
     }

graphlit_api/query_agents.py

@@ -7,6 +7,7 @@
 
 from .base_model import BaseModel
 from .enums import (
+    AgentChannelTypes,
     AgentTypes,
     ContentTypes,
     EntityState,
@@ -40,6 +41,7 @@ class QueryAgentsAgentsResults(BaseModel):
     schedule_policy: Optional["QueryAgentsAgentsResultsSchedulePolicy"] = Field(
         alias="schedulePolicy"
     )
+    channels: Optional[list["QueryAgentsAgentsResultsChannels"]]
     timeout: Optional[Any]
     scratchpad: Optional[str]
 
@@ -214,6 +216,13 @@ class QueryAgentsAgentsResultsSchedulePolicy(BaseModel):
     time_zone_id: Optional[str] = Field(alias="timeZoneId")
 
 
+class QueryAgentsAgentsResultsChannels(BaseModel):
+    type: AgentChannelTypes
+    identifier: str
+    instructions: Optional[str]
+    label: Optional[str]
+
+
 QueryAgents.model_rebuild()
 QueryAgentsAgents.model_rebuild()
 QueryAgentsAgentsResults.model_rebuild()