graphlit-client 1.0.20250627003__tar.gz → 1.0.20250627004__tar.gz

{graphlit_client-1.0.20250627003/graphlit_client.egg-info → graphlit_client-1.0.20250627004}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: graphlit-client
-Version: 1.0.20250627003
+Version: 1.0.20250627004
 Summary: Graphlit API Python Client
 Home-page: https://github.com/graphlit/graphlit-client-python
 Author: Unstruk Data Inc.

{graphlit_client-1.0.20250627003 → graphlit_client-1.0.20250627004}/graphlit_api/__init__.py

@@ -1931,6 +1931,7 @@ from .operations import (
     QUERY_WORKFLOWS_GQL,
     REMOVE_CONTENTS_FROM_COLLECTION_GQL,
     RETRIEVE_SOURCES_GQL,
+    RETRIEVE_VIEW_GQL,
     REVISE_CONTENT_GQL,
     REVISE_ENCODED_IMAGE_GQL,
     REVISE_IMAGE_GQL,
@@ -2667,6 +2668,12 @@ from .retrieve_sources import (
     RetrieveSourcesRetrieveSourcesResults,
     RetrieveSourcesRetrieveSourcesResultsContent,
 )
+from .retrieve_view import (
+    RetrieveView,
+    RetrieveViewRetrieveView,
+    RetrieveViewRetrieveViewResults,
+    RetrieveViewRetrieveViewResultsContent,
+)
 from .revise_content import (
     ReviseContent,
     ReviseContentReviseContent,
@@ -5350,6 +5357,7 @@ __all__ = [
     "QueryWorkflowsWorkflowsResultsStoragePolicy",
     "REMOVE_CONTENTS_FROM_COLLECTION_GQL",
     "RETRIEVE_SOURCES_GQL",
+    "RETRIEVE_VIEW_GQL",
     "REVISE_CONTENT_GQL",
     "REVISE_ENCODED_IMAGE_GQL",
     "REVISE_IMAGE_GQL",
@@ -5389,6 +5397,10 @@ __all__ = [
     "RetrieveSourcesRetrieveSources",
     "RetrieveSourcesRetrieveSourcesResults",
     "RetrieveSourcesRetrieveSourcesResultsContent",
+    "RetrieveView",
+    "RetrieveViewRetrieveView",
+    "RetrieveViewRetrieveViewResults",
+    "RetrieveViewRetrieveViewResultsContent",
     "ReviseContent",
     "ReviseContentReviseContent",
     "ReviseContentReviseContentConversation",

{graphlit_client-1.0.20250627003 → graphlit_client-1.0.20250627004}/graphlit_api/client.py

@@ -612,6 +612,7 @@ from .operations import (
     QUERY_WORKFLOWS_GQL,
     REMOVE_CONTENTS_FROM_COLLECTION_GQL,
     RETRIEVE_SOURCES_GQL,
+    RETRIEVE_VIEW_GQL,
     REVISE_CONTENT_GQL,
     REVISE_ENCODED_IMAGE_GQL,
     REVISE_IMAGE_GQL,
@@ -722,6 +723,7 @@ from .query_views import QueryViews
 from .query_workflows import QueryWorkflows
 from .remove_contents_from_collection import RemoveContentsFromCollection
 from .retrieve_sources import RetrieveSources
+from .retrieve_view import RetrieveView
 from .revise_content import ReviseContent
 from .revise_encoded_image import ReviseEncodedImage
 from .revise_image import ReviseImage
@@ -2357,6 +2359,31 @@ class Client(AsyncBaseClient):
         data = self.get_data(response)
         return RetrieveSources.model_validate(data)
 
+    async def retrieve_view(
+        self,
+        prompt: str,
+        id: str,
+        retrieval_strategy: Union[Optional[RetrievalStrategyInput], UnsetType] = UNSET,
+        reranking_strategy: Union[Optional[RerankingStrategyInput], UnsetType] = UNSET,
+        correlation_id: Union[Optional[str], UnsetType] = UNSET,
+        **kwargs: Any
+    ) -> RetrieveView:
+        variables: Dict[str, object] = {
+            "prompt": prompt,
+            "id": id,
+            "retrievalStrategy": retrieval_strategy,
+            "rerankingStrategy": reranking_strategy,
+            "correlationId": correlation_id,
+        }
+        response = await self.execute(
+            query=RETRIEVE_VIEW_GQL,
+            operation_name="RetrieveView",
+            variables=variables,
+            **kwargs
+        )
+        data = self.get_data(response)
+        return RetrieveView.model_validate(data)
+
     async def revise_content(
         self,
         prompt: str,

{graphlit_client-1.0.20250627003 → graphlit_client-1.0.20250627004}/graphlit_api/exceptions.py

@@ -15,7 +15,7 @@ class GraphQLClientHttpError(GraphQLClientError):
         self.response = response
 
     def __str__(self) -> str:
-        return f"HTTP status code: {self.status_code}"
+        return f"HTTP status code: {self.status_code}\n{self.response.text}"
 
 
 class GraphQLClientInvalidResponseError(GraphQLClientError):

{graphlit_client-1.0.20250627003 → graphlit_client-1.0.20250627004}/graphlit_api/operations.py

@@ -273,6 +273,7 @@ __all__ = [
     "QUERY_WORKFLOWS_GQL",
     "REMOVE_CONTENTS_FROM_COLLECTION_GQL",
     "RETRIEVE_SOURCES_GQL",
+    "RETRIEVE_VIEW_GQL",
     "REVISE_CONTENT_GQL",
     "REVISE_ENCODED_IMAGE_GQL",
     "REVISE_IMAGE_GQL",
@@ -5942,6 +5943,32 @@ mutation RetrieveSources($prompt: String!, $filter: ContentFilter, $augmentedFil
 }
 """
 
+RETRIEVE_VIEW_GQL = """
+mutation RetrieveView($prompt: String!, $id: ID!, $retrievalStrategy: RetrievalStrategyInput, $rerankingStrategy: RerankingStrategyInput, $correlationId: String) {
+  retrieveView(
+    prompt: $prompt
+    id: $id
+    retrievalStrategy: $retrievalStrategy
+    rerankingStrategy: $rerankingStrategy
+    correlationId: $correlationId
+  ) {
+    results {
+      type
+      content {
+        id
+      }
+      text
+      metadata
+      relevance
+      startTime
+      endTime
+      pageNumber
+      frameNumber
+    }
+  }
+}
+"""
+
 REVISE_CONTENT_GQL = """
 mutation ReviseContent($prompt: String!, $content: EntityReferenceInput!, $id: ID, $specification: EntityReferenceInput, $correlationId: String) {
   reviseContent(

graphlit_client-1.0.20250627004/graphlit_api/retrieve_view.py

@@ -0,0 +1,38 @@
+# Generated by ariadne-codegen
+# Source: ./documents
+
+from typing import Any, List, Optional
+
+from pydantic import Field
+
+from .base_model import BaseModel
+from .enums import ContentSourceTypes
+
+
+class RetrieveView(BaseModel):
+    retrieve_view: Optional["RetrieveViewRetrieveView"] = Field(alias="retrieveView")
+
+
+class RetrieveViewRetrieveView(BaseModel):
+    results: Optional[List[Optional["RetrieveViewRetrieveViewResults"]]]
+
+
+class RetrieveViewRetrieveViewResults(BaseModel):
+    type: Optional[ContentSourceTypes]
+    content: "RetrieveViewRetrieveViewResultsContent"
+    text: str
+    metadata: Optional[str]
+    relevance: float
+    start_time: Optional[Any] = Field(alias="startTime")
+    end_time: Optional[Any] = Field(alias="endTime")
+    page_number: Optional[int] = Field(alias="pageNumber")
+    frame_number: Optional[int] = Field(alias="frameNumber")
+
+
+class RetrieveViewRetrieveViewResultsContent(BaseModel):
+    id: str
+
+
+RetrieveView.model_rebuild()
+RetrieveViewRetrieveView.model_rebuild()
+RetrieveViewRetrieveViewResults.model_rebuild()

{graphlit_client-1.0.20250627003 → graphlit_client-1.0.20250627004/graphlit_client.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: graphlit-client
-Version: 1.0.20250627003
+Version: 1.0.20250627004
 Summary: Graphlit API Python Client
 Home-page: https://github.com/graphlit/graphlit-client-python
 Author: Unstruk Data Inc.

{graphlit_client-1.0.20250627003 → graphlit_client-1.0.20250627004}/graphlit_client.egg-info/SOURCES.txt

@@ -283,6 +283,7 @@ graphlit_api/query_views.py
 graphlit_api/query_workflows.py
 graphlit_api/remove_contents_from_collection.py
 graphlit_api/retrieve_sources.py
+graphlit_api/retrieve_view.py
 graphlit_api/revise_content.py
 graphlit_api/revise_encoded_image.py
 graphlit_api/revise_image.py