Updated to latest API release.

graphlit · Dec 7, 2024 · 4805a61 · 4805a61
1 parent 558f51c
commit 4805a61
Show file tree

Hide file tree

Showing 6 changed files with 8 additions and 8 deletions.
diff --git a/documents/content/ExtractContents.graphql b/documents/content/ExtractContents.graphql
@@ -1,4 +1,4 @@
-mutation ExtractContents($prompt: String!, $filter: ContentFilter, $specification: EntityReferenceInput!, $tools: [ToolDefinitionInput!]!, $correlationId: String) {
+mutation ExtractContents($prompt: String!, $filter: ContentFilter, $specification: EntityReferenceInput, $tools: [ToolDefinitionInput!]!, $correlationId: String) {
   extractContents(prompt: $prompt, filter: $filter, specification: $specification, tools: $tools, correlationId: $correlationId) {
     specification {
       id

diff --git a/documents/content/ExtractText.graphql b/documents/content/ExtractText.graphql
@@ -1,4 +1,4 @@
-mutation ExtractText($prompt: String!, $text: String!, $textType: TextTypes, $specification: EntityReferenceInput!, $tools: [ToolDefinitionInput!]!, $correlationId: String) {
+mutation ExtractText($prompt: String!, $text: String!, $textType: TextTypes, $specification: EntityReferenceInput, $tools: [ToolDefinitionInput!]!, $correlationId: String) {
   extractText(prompt: $prompt, text: $text, textType: $textType, specification: $specification, tools: $tools, correlationId: $correlationId) {
     specification {
       id

diff --git a/graphlit_api/client.py b/graphlit_api/client.py
@@ -1161,9 +1161,9 @@ async def describe_image(
     async def extract_contents(
         self,
         prompt: str,
-        specification: EntityReferenceInput,
         tools: List[ToolDefinitionInput],
         filter: Union[Optional[ContentFilter], UnsetType] = UNSET,
+        specification: Union[Optional[EntityReferenceInput], UnsetType] = UNSET,
         correlation_id: Union[Optional[str], UnsetType] = UNSET,
         **kwargs: Any
     ) -> ExtractContents:
@@ -1187,9 +1187,9 @@ async def extract_text(
         self,
         prompt: str,
         text: str,
-        specification: EntityReferenceInput,
         tools: List[ToolDefinitionInput],
         text_type: Union[Optional[TextTypes], UnsetType] = UNSET,
+        specification: Union[Optional[EntityReferenceInput], UnsetType] = UNSET,
         correlation_id: Union[Optional[str], UnsetType] = UNSET,
         **kwargs: Any
     ) -> ExtractText:

diff --git a/graphlit_api/extract_contents.py b/graphlit_api/extract_contents.py
@@ -15,7 +15,7 @@ class ExtractContents(BaseModel):
 
 
 class ExtractContentsExtractContents(BaseModel):
-    specification: "ExtractContentsExtractContentsSpecification"
+    specification: Optional["ExtractContentsExtractContentsSpecification"]
     content: Optional["ExtractContentsExtractContentsContent"]
     name: str
     value: str

diff --git a/graphlit_api/extract_text.py b/graphlit_api/extract_text.py
@@ -15,7 +15,7 @@ class ExtractText(BaseModel):
 
 
 class ExtractTextExtractText(BaseModel):
-    specification: "ExtractTextExtractTextSpecification"
+    specification: Optional["ExtractTextExtractTextSpecification"]
     content: Optional["ExtractTextExtractTextContent"]
     name: str
     value: str

diff --git a/graphlit_api/operations.py b/graphlit_api/operations.py
@@ -1090,7 +1090,7 @@
 """
 
 EXTRACT_CONTENTS_GQL = """
-mutation ExtractContents($prompt: String!, $filter: ContentFilter, $specification: EntityReferenceInput!, $tools: [ToolDefinitionInput!]!, $correlationId: String) {
+mutation ExtractContents($prompt: String!, $filter: ContentFilter, $specification: EntityReferenceInput, $tools: [ToolDefinitionInput!]!, $correlationId: String) {
   extractContents(
     prompt: $prompt
     filter: $filter
@@ -1115,7 +1115,7 @@
 """
 
 EXTRACT_TEXT_GQL = """
-mutation ExtractText($prompt: String!, $text: String!, $textType: TextTypes, $specification: EntityReferenceInput!, $tools: [ToolDefinitionInput!]!, $correlationId: String) {
+mutation ExtractText($prompt: String!, $text: String!, $textType: TextTypes, $specification: EntityReferenceInput, $tools: [ToolDefinitionInput!]!, $correlationId: String) {
   extractText(
     prompt: $prompt
     text: $text