Updated to latest API release.

Author: kirk-marple

documents/agent/GetAgent.graphql

@@ -32,6 +32,9 @@ query GetAgent($id: ID!, $correlationId: String) {
       instructions
       label
     }
+    connectors {
+      id
+    }
     timeout
     prompt
     scratchpad

documents/agent/QueryAgents.graphql

@@ -34,6 +34,9 @@ query QueryAgents($filter: AgentFilter, $correlationId: String) {
         instructions
         label
       }
+      connectors {
+        id
+      }
       timeout
       prompt
       scratchpad

documents/content/AddContentLabel.graphql

@@ -0,0 +1,6 @@
+mutation AddContentLabel($id: ID!, $label: String!) {
+  addContentLabel(id: $id, label: $label) {
+    id
+    name
+  }
+}

documents/content/RemoveContentLabel.graphql

@@ -0,0 +1,6 @@
+mutation RemoveContentLabel($id: ID!, $label: String!) {
+  removeContentLabel(id: $id, label: $label) {
+    id
+    name
+  }
+}

graphlit_api/__init__.py

@@ -1,5 +1,6 @@
 # Generated by ariadne-codegen
 
+from .add_content_label import AddContentLabel, AddContentLabelAddContentLabel
 from .add_contents_to_collections import (
     AddContentsToCollections,
     AddContentsToCollectionsAddContentsToCollections,
@@ -971,6 +972,7 @@
     GetAgent,
     GetAgentAgent,
     GetAgentAgentChannels,
+    GetAgentAgentConnectors,
     GetAgentAgentFilter,
     GetAgentAgentFilterFeeds,
     GetAgentAgentOwner,
@@ -2492,6 +2494,7 @@
     MatchEntityMatchEntityReferenceObservable,
 )
 from .operations import (
+    ADD_CONTENT_LABEL_GQL,
     ADD_CONTENTS_TO_COLLECTIONS_GQL,
     ADD_CONVERSATIONS_TO_COLLECTIONS_GQL,
     ADD_SKILLS_TO_COLLECTIONS_GQL,
@@ -2879,6 +2882,7 @@
     QUERY_VIEWS_GQL,
     QUERY_WORKFLOWS_GQL,
     REJECT_CONTENT_GQL,
+    REMOVE_CONTENT_LABEL_GQL,
     REMOVE_CONTENTS_FROM_COLLECTION_GQL,
     REMOVE_CONVERSATIONS_FROM_COLLECTION_GQL,
     REMOVE_SKILLS_FROM_COLLECTION_GQL,
@@ -3061,6 +3065,7 @@
     QueryAgentsAgents,
     QueryAgentsAgentsResults,
     QueryAgentsAgentsResultsChannels,
+    QueryAgentsAgentsResultsConnectors,
     QueryAgentsAgentsResultsFilter,
     QueryAgentsAgentsResultsFilterFeeds,
     QueryAgentsAgentsResultsOwner,
@@ -4547,6 +4552,10 @@
     QueryWorkflowsWorkflowsResultsStoragePolicy,
 )
 from .reject_content import RejectContent, RejectContentRejectContent
+from .remove_content_label import (
+    RemoveContentLabel,
+    RemoveContentLabelRemoveContentLabel,
+)
 from .remove_contents_from_collection import (
     RemoveContentsFromCollection,
     RemoveContentsFromCollectionRemoveContentsFromCollection,
@@ -5075,10 +5084,13 @@
 
 __all__ = [
     "ADD_CONTENTS_TO_COLLECTIONS_GQL",
+    "ADD_CONTENT_LABEL_GQL",
     "ADD_CONVERSATIONS_TO_COLLECTIONS_GQL",
     "ADD_SKILLS_TO_COLLECTIONS_GQL",
     "APPROVE_CONTENT_GQL",
     "ASK_GRAPHLIT_GQL",
+    "AddContentLabel",
+    "AddContentLabelAddContentLabel",
     "AddContentsToCollections",
     "AddContentsToCollectionsAddContentsToCollections",
     "AddContentsToCollectionsAddContentsToCollectionsContents",
@@ -6281,6 +6293,7 @@
     "GetAgent",
     "GetAgentAgent",
     "GetAgentAgentChannels",
+    "GetAgentAgentConnectors",
     "GetAgentAgentFilter",
     "GetAgentAgentFilterFeeds",
     "GetAgentAgentOwner",
@@ -7738,6 +7751,7 @@
     "QueryAgentsAgents",
     "QueryAgentsAgentsResults",
     "QueryAgentsAgentsResultsChannels",
+    "QueryAgentsAgentsResultsConnectors",
     "QueryAgentsAgentsResultsFilter",
     "QueryAgentsAgentsResultsFilterFeeds",
     "QueryAgentsAgentsResultsOwner",
@@ -9034,6 +9048,7 @@
     "QuiverImagePublishingPropertiesInput",
     "REJECT_CONTENT_GQL",
     "REMOVE_CONTENTS_FROM_COLLECTION_GQL",
+    "REMOVE_CONTENT_LABEL_GQL",
     "REMOVE_CONVERSATIONS_FROM_COLLECTION_GQL",
     "REMOVE_SKILLS_FROM_COLLECTION_GQL",
     "RESEARCH_CONTENTS_GQL",
@@ -9064,6 +9079,8 @@
     "RejectContent",
     "RejectContentRejectContent",
     "RelationshipDirections",
+    "RemoveContentLabel",
+    "RemoveContentLabelRemoveContentLabel",
     "RemoveContentsFromCollection",
     "RemoveContentsFromCollectionRemoveContentsFromCollection",
     "RemoveContentsFromCollectionRemoveContentsFromCollectionContents",

graphlit_api/add_content_label.py

@@ -0,0 +1,22 @@
+# Generated by ariadne-codegen
+# Source: ./documents
+
+from typing import Optional
+
+from pydantic import Field
+
+from .base_model import BaseModel
+
+
+class AddContentLabel(BaseModel):
+    add_content_label: Optional["AddContentLabelAddContentLabel"] = Field(
+        alias="addContentLabel"
+    )
+
+
+class AddContentLabelAddContentLabel(BaseModel):
+    id: str
+    name: str
+
+
+AddContentLabel.model_rebuild()

graphlit_api/client.py

@@ -3,6 +3,7 @@
 
 from typing import Any, Optional, Union
 
+from .add_content_label import AddContentLabel
 from .add_contents_to_collections import AddContentsToCollections
 from .add_conversations_to_collections import AddConversationsToCollections
 from .add_skills_to_collections import AddSkillsToCollections
@@ -460,6 +461,7 @@
 from .map_web import MapWeb
 from .match_entity import MatchEntity
 from .operations import (
+    ADD_CONTENT_LABEL_GQL,
     ADD_CONTENTS_TO_COLLECTIONS_GQL,
     ADD_CONVERSATIONS_TO_COLLECTIONS_GQL,
     ADD_SKILLS_TO_COLLECTIONS_GQL,
@@ -847,6 +849,7 @@
     QUERY_VIEWS_GQL,
     QUERY_WORKFLOWS_GQL,
     REJECT_CONTENT_GQL,
+    REMOVE_CONTENT_LABEL_GQL,
     REMOVE_CONTENTS_FROM_COLLECTION_GQL,
     REMOVE_CONVERSATIONS_FROM_COLLECTION_GQL,
     REMOVE_SKILLS_FROM_COLLECTION_GQL,
@@ -1042,6 +1045,7 @@
 from .query_views import QueryViews
 from .query_workflows import QueryWorkflows
 from .reject_content import RejectContent
+from .remove_content_label import RemoveContentLabel
 from .remove_contents_from_collection import RemoveContentsFromCollection
 from .remove_conversations_from_collection import RemoveConversationsFromCollection
 from .remove_skills_from_collection import RemoveSkillsFromCollection
@@ -1916,6 +1920,19 @@ async def update_connector(
         data = self.get_data(response)
         return UpdateConnector.model_validate(data)
 
+    async def add_content_label(
+        self, id: str, label: str, **kwargs: Any
+    ) -> AddContentLabel:
+        variables: dict[str, object] = {"id": id, "label": label}
+        response = await self.execute(
+            query=ADD_CONTENT_LABEL_GQL,
+            operation_name="AddContentLabel",
+            variables=variables,
+            **kwargs
+        )
+        data = self.get_data(response)
+        return AddContentLabel.model_validate(data)
+
     async def approve_content(self, id: str, **kwargs: Any) -> ApproveContent:
         variables: dict[str, object] = {"id": id}
         response = await self.execute(
@@ -2642,6 +2659,19 @@ async def reject_content(
         data = self.get_data(response)
         return RejectContent.model_validate(data)
 
+    async def remove_content_label(
+        self, id: str, label: str, **kwargs: Any
+    ) -> RemoveContentLabel:
+        variables: dict[str, object] = {"id": id, "label": label}
+        response = await self.execute(
+            query=REMOVE_CONTENT_LABEL_GQL,
+            operation_name="RemoveContentLabel",
+            variables=variables,
+            **kwargs
+        )
+        data = self.get_data(response)
+        return RemoveContentLabel.model_validate(data)
+
     async def research_contents(
         self,
         connector: ContentPublishingConnectorInput,

graphlit_api/get_agent.py

@@ -35,6 +35,7 @@ class GetAgentAgent(BaseModel):
         alias="schedulePolicy"
     )
     channels: Optional[list["GetAgentAgentChannels"]]
+    connectors: Optional[list["GetAgentAgentConnectors"]]
     timeout: Optional[Any]
     prompt: Optional[str]
     scratchpad: Optional[str]
@@ -74,6 +75,10 @@ class GetAgentAgentChannels(BaseModel):
     label: Optional[str]
 
 
+class GetAgentAgentConnectors(BaseModel):
+    id: str
+
+
 GetAgent.model_rebuild()
 GetAgentAgent.model_rebuild()
 GetAgentAgentFilter.model_rebuild()

graphlit_api/input_types.py

@@ -2825,6 +2825,7 @@ class AgentInput(BaseModel):
     prompt: Optional[str] = None
     scratchpad: Optional[str] = None
     channels: Optional[list["AgentChannelInput"]] = None
+    connectors: Optional[list["EntityReferenceInput"]] = None
 
 
 class CrustdataPersonDiscoveryFilterInput(BaseModel):
@@ -7034,6 +7035,7 @@ class AgentUpdateInput(BaseModel):
     prompt: Optional[str] = None
     scratchpad: Optional[str] = None
     channels: Optional[list["AgentChannelInput"]] = None
+    connectors: Optional[list["EntityReferenceInput"]] = None
 
 
 class IssueFeedPropertiesInput(BaseModel):

graphlit_api/operations.py

@@ -3,6 +3,7 @@
 
 __all__ = [
     "ADD_CONTENTS_TO_COLLECTIONS_GQL",
+    "ADD_CONTENT_LABEL_GQL",
     "ADD_CONVERSATIONS_TO_COLLECTIONS_GQL",
     "ADD_SKILLS_TO_COLLECTIONS_GQL",
     "APPROVE_CONTENT_GQL",
@@ -390,6 +391,7 @@
     "QUERY_WORKFLOWS_GQL",
     "REJECT_CONTENT_GQL",
     "REMOVE_CONTENTS_FROM_COLLECTION_GQL",
+    "REMOVE_CONTENT_LABEL_GQL",
     "REMOVE_CONVERSATIONS_FROM_COLLECTION_GQL",
     "REMOVE_SKILLS_FROM_COLLECTION_GQL",
     "RESEARCH_CONTENTS_GQL",
@@ -572,6 +574,9 @@
       instructions
       label
     }
+    connectors {
+      id
+    }
     timeout
     prompt
     scratchpad
@@ -616,6 +621,9 @@
         instructions
         label
       }
+      connectors {
+        id
+      }
       timeout
       prompt
       scratchpad
@@ -1618,6 +1626,15 @@
 }
 """
 
+ADD_CONTENT_LABEL_GQL = """
+mutation AddContentLabel($id: ID!, $label: String!) {
+  addContentLabel(id: $id, label: $label) {
+    id
+    name
+  }
+}
+"""
+
 APPROVE_CONTENT_GQL = """
 mutation ApproveContent($id: ID!) {
   approveContent(id: $id) {
@@ -4749,6 +4766,15 @@
 }
 """
 
+REMOVE_CONTENT_LABEL_GQL = """
+mutation RemoveContentLabel($id: ID!, $label: String!) {
+  removeContentLabel(id: $id, label: $label) {
+    id
+    name
+  }
+}
+"""
+
 RESEARCH_CONTENTS_GQL = """
 mutation ResearchContents($connector: ContentPublishingConnectorInput!, $filter: ContentFilter, $name: String, $summarySpecification: EntityReferenceInput, $publishSpecification: EntityReferenceInput, $workflow: EntityReferenceInput, $correlationId: String) {
   researchContents(