Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
6 changes: 2 additions & 4 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@ name = "robosystems-client"
version = "0.4.0"
description = "Python client library for RoboSystems Financial Knowledge Graph API"
readme = "README.md"
requires-python = ">=3.10"
requires-python = ">=3.11"
authors = [
{name = "RFS LLC"},
]
Expand All @@ -20,7 +20,6 @@ classifiers = [
"Natural Language :: English",
"Operating System :: OS Independent",
"Programming Language :: Python :: 3",
"Programming Language :: Python :: 3.10",
"Programming Language :: Python :: 3.11",
"Programming Language :: Python :: 3.12",
"Programming Language :: Python :: 3.13",
Expand All @@ -36,12 +35,11 @@ dependencies = [
"pydantic>=2.11.7",
"typing-extensions>=4.0.0",
"attrs>=23.0.0",
"python-dateutil>=2.8.0",
]

[project.optional-dependencies]
dev = [
"openapi-python-client>=0.21.8",
"openapi-python-client>=0.29.0",
"basedpyright>=1.21.0",
"pytest-asyncio>=0.26.0",
"pytest>=8.3.5",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -109,7 +109,7 @@ def sync_detailed(
body: FileUploadRequest,
idempotency_key: None | str | Unset = UNSET,
) -> Response[ErrorResponse | OperationEnvelope]:
"""Create File Upload (presign an S3 upload)
"""Presign a File Upload

Presign an S3 URL for direct upload and register the file. After uploading to the returned URL, call
`POST /operations/ingest-file` to stage it into DuckDB. The staging table is auto-created if
Expand Down Expand Up @@ -151,7 +151,7 @@ def sync(
body: FileUploadRequest,
idempotency_key: None | str | Unset = UNSET,
) -> ErrorResponse | OperationEnvelope | None:
"""Create File Upload (presign an S3 upload)
"""Presign a File Upload

Presign an S3 URL for direct upload and register the file. After uploading to the returned URL, call
`POST /operations/ingest-file` to stage it into DuckDB. The staging table is auto-created if
Expand Down Expand Up @@ -188,7 +188,7 @@ async def asyncio_detailed(
body: FileUploadRequest,
idempotency_key: None | str | Unset = UNSET,
) -> Response[ErrorResponse | OperationEnvelope]:
"""Create File Upload (presign an S3 upload)
"""Presign a File Upload

Presign an S3 URL for direct upload and register the file. After uploading to the returned URL, call
`POST /operations/ingest-file` to stage it into DuckDB. The staging table is auto-created if
Expand Down Expand Up @@ -228,7 +228,7 @@ async def asyncio(
body: FileUploadRequest,
idempotency_key: None | str | Unset = UNSET,
) -> ErrorResponse | OperationEnvelope | None:
"""Create File Upload (presign an S3 upload)
"""Presign a File Upload

Presign an S3 URL for direct upload and register the file. After uploading to the returned URL, call
`POST /operations/ingest-file` to stage it into DuckDB. The staging table is auto-created if
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -109,7 +109,7 @@ def sync_detailed(
body: DeleteDocumentOp,
idempotency_key: None | str | Unset = UNSET,
) -> Response[ErrorResponse | OperationEnvelope]:
"""Delete Document (remove from the corpus)
"""Delete a Document

Delete a document from PostgreSQL and OpenSearch by id.

Expand Down Expand Up @@ -149,7 +149,7 @@ def sync(
body: DeleteDocumentOp,
idempotency_key: None | str | Unset = UNSET,
) -> ErrorResponse | OperationEnvelope | None:
"""Delete Document (remove from the corpus)
"""Delete a Document

Delete a document from PostgreSQL and OpenSearch by id.

Expand Down Expand Up @@ -184,7 +184,7 @@ async def asyncio_detailed(
body: DeleteDocumentOp,
idempotency_key: None | str | Unset = UNSET,
) -> Response[ErrorResponse | OperationEnvelope]:
"""Delete Document (remove from the corpus)
"""Delete a Document

Delete a document from PostgreSQL and OpenSearch by id.

Expand Down Expand Up @@ -222,7 +222,7 @@ async def asyncio(
body: DeleteDocumentOp,
idempotency_key: None | str | Unset = UNSET,
) -> ErrorResponse | OperationEnvelope | None:
"""Delete Document (remove from the corpus)
"""Delete a Document

Delete a document from PostgreSQL and OpenSearch by id.

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -109,7 +109,7 @@ def sync_detailed(
body: DeleteFileOp,
idempotency_key: None | str | Unset = UNSET,
) -> Response[ErrorResponse | OperationEnvelope]:
"""Delete File
"""Delete a File

Delete a file from S3 and PostgreSQL. `cascade=true` also removes its rows from DuckDB staging
tables and marks the graph stale.
Expand Down Expand Up @@ -150,7 +150,7 @@ def sync(
body: DeleteFileOp,
idempotency_key: None | str | Unset = UNSET,
) -> ErrorResponse | OperationEnvelope | None:
"""Delete File
"""Delete a File

Delete a file from S3 and PostgreSQL. `cascade=true` also removes its rows from DuckDB staging
tables and marks the graph stale.
Expand Down Expand Up @@ -186,7 +186,7 @@ async def asyncio_detailed(
body: DeleteFileOp,
idempotency_key: None | str | Unset = UNSET,
) -> Response[ErrorResponse | OperationEnvelope]:
"""Delete File
"""Delete a File

Delete a file from S3 and PostgreSQL. `cascade=true` also removes its rows from DuckDB staging
tables and marks the graph stale.
Expand Down Expand Up @@ -225,7 +225,7 @@ async def asyncio(
body: DeleteFileOp,
idempotency_key: None | str | Unset = UNSET,
) -> ErrorResponse | OperationEnvelope | None:
"""Delete File
"""Delete a File

Delete a file from S3 and PostgreSQL. `cascade=true` also removes its rows from DuckDB staging
tables and marks the graph stale.
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -109,7 +109,7 @@ def sync_detailed(
body: ForgetOp,
idempotency_key: None | str | Unset = UNSET,
) -> Response[ErrorResponse | OperationEnvelope]:
"""Forget (delete a semantic memory)
"""Delete a Semantic Memory

Delete a semantic memory by its server-generated id.

Expand Down Expand Up @@ -149,7 +149,7 @@ def sync(
body: ForgetOp,
idempotency_key: None | str | Unset = UNSET,
) -> ErrorResponse | OperationEnvelope | None:
"""Forget (delete a semantic memory)
"""Delete a Semantic Memory

Delete a semantic memory by its server-generated id.

Expand Down Expand Up @@ -184,7 +184,7 @@ async def asyncio_detailed(
body: ForgetOp,
idempotency_key: None | str | Unset = UNSET,
) -> Response[ErrorResponse | OperationEnvelope]:
"""Forget (delete a semantic memory)
"""Delete a Semantic Memory

Delete a semantic memory by its server-generated id.

Expand Down Expand Up @@ -222,7 +222,7 @@ async def asyncio(
body: ForgetOp,
idempotency_key: None | str | Unset = UNSET,
) -> ErrorResponse | OperationEnvelope | None:
"""Forget (delete a semantic memory)
"""Delete a Semantic Memory

Delete a semantic memory by its server-generated id.

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -109,7 +109,7 @@ def sync_detailed(
body: IndexDocumentOp,
idempotency_key: None | str | Unset = UNSET,
) -> Response[ErrorResponse | OperationEnvelope]:
"""Index Document (write to the corpus)
"""Index a Document

Create a document (omit `document_id`) or update one (provide it). Stored in PostgreSQL, synced to
OpenSearch for search.
Expand Down Expand Up @@ -153,7 +153,7 @@ def sync(
body: IndexDocumentOp,
idempotency_key: None | str | Unset = UNSET,
) -> ErrorResponse | OperationEnvelope | None:
"""Index Document (write to the corpus)
"""Index a Document

Create a document (omit `document_id`) or update one (provide it). Stored in PostgreSQL, synced to
OpenSearch for search.
Expand Down Expand Up @@ -192,7 +192,7 @@ async def asyncio_detailed(
body: IndexDocumentOp,
idempotency_key: None | str | Unset = UNSET,
) -> Response[ErrorResponse | OperationEnvelope]:
"""Index Document (write to the corpus)
"""Index a Document

Create a document (omit `document_id`) or update one (provide it). Stored in PostgreSQL, synced to
OpenSearch for search.
Expand Down Expand Up @@ -234,7 +234,7 @@ async def asyncio(
body: IndexDocumentOp,
idempotency_key: None | str | Unset = UNSET,
) -> ErrorResponse | OperationEnvelope | None:
"""Index Document (write to the corpus)
"""Index a Document

Create a document (omit `document_id`) or update one (provide it). Stored in PostgreSQL, synced to
OpenSearch for search.
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -109,7 +109,7 @@ def sync_detailed(
body: IngestFileOp,
idempotency_key: None | str | Unset = UNSET,
) -> Response[ErrorResponse | OperationEnvelope]:
"""Ingest File (stage an uploaded file)
"""Stage an Uploaded File

Mark an uploaded file ready and stage it into DuckDB. Small files stage directly (sync); large files
stage via a background job (returns a pending envelope with an `operation_id` to monitor). Set
Expand Down Expand Up @@ -154,7 +154,7 @@ def sync(
body: IngestFileOp,
idempotency_key: None | str | Unset = UNSET,
) -> ErrorResponse | OperationEnvelope | None:
"""Ingest File (stage an uploaded file)
"""Stage an Uploaded File

Mark an uploaded file ready and stage it into DuckDB. Small files stage directly (sync); large files
stage via a background job (returns a pending envelope with an `operation_id` to monitor). Set
Expand Down Expand Up @@ -194,7 +194,7 @@ async def asyncio_detailed(
body: IngestFileOp,
idempotency_key: None | str | Unset = UNSET,
) -> Response[ErrorResponse | OperationEnvelope]:
"""Ingest File (stage an uploaded file)
"""Stage an Uploaded File

Mark an uploaded file ready and stage it into DuckDB. Small files stage directly (sync); large files
stage via a background job (returns a pending envelope with an `operation_id` to monitor). Set
Expand Down Expand Up @@ -237,7 +237,7 @@ async def asyncio(
body: IngestFileOp,
idempotency_key: None | str | Unset = UNSET,
) -> ErrorResponse | OperationEnvelope | None:
"""Ingest File (stage an uploaded file)
"""Stage an Uploaded File

Mark an uploaded file ready and stage it into DuckDB. Small files stage directly (sync); large files
stage via a background job (returns a pending envelope with an `operation_id` to monitor). Set
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -109,7 +109,7 @@ def sync_detailed(
body: RememberOp,
idempotency_key: None | str | Unset = UNSET,
) -> Response[ErrorResponse | OperationEnvelope]:
"""Remember (write semantic memory)
"""Store a Semantic Memory

Store a semantic memory in the graph's per-graph memory store. The text is embedded locally; recall
it later via `POST /memory/recall`.
Expand Down Expand Up @@ -150,7 +150,7 @@ def sync(
body: RememberOp,
idempotency_key: None | str | Unset = UNSET,
) -> ErrorResponse | OperationEnvelope | None:
"""Remember (write semantic memory)
"""Store a Semantic Memory

Store a semantic memory in the graph's per-graph memory store. The text is embedded locally; recall
it later via `POST /memory/recall`.
Expand Down Expand Up @@ -186,7 +186,7 @@ async def asyncio_detailed(
body: RememberOp,
idempotency_key: None | str | Unset = UNSET,
) -> Response[ErrorResponse | OperationEnvelope]:
"""Remember (write semantic memory)
"""Store a Semantic Memory

Store a semantic memory in the graph's per-graph memory store. The text is embedded locally; recall
it later via `POST /memory/recall`.
Expand Down Expand Up @@ -225,7 +225,7 @@ async def asyncio(
body: RememberOp,
idempotency_key: None | str | Unset = UNSET,
) -> ErrorResponse | OperationEnvelope | None:
"""Remember (write semantic memory)
"""Store a Semantic Memory

Store a semantic memory in the graph's per-graph memory store. The text is embedded locally; recall
it later via `POST /memory/recall`.
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -109,7 +109,7 @@ def sync_detailed(
body: UpdateMemoryOp,
idempotency_key: None | str | Unset = UNSET,
) -> Response[ErrorResponse | OperationEnvelope]:
"""Update Memory (edit a semantic memory)
"""Update a Semantic Memory

Partially update a stored memory by its server-generated id. Only supplied fields change; the memory
is re-embedded when `text` changes.
Expand Down Expand Up @@ -154,7 +154,7 @@ def sync(
body: UpdateMemoryOp,
idempotency_key: None | str | Unset = UNSET,
) -> ErrorResponse | OperationEnvelope | None:
"""Update Memory (edit a semantic memory)
"""Update a Semantic Memory

Partially update a stored memory by its server-generated id. Only supplied fields change; the memory
is re-embedded when `text` changes.
Expand Down Expand Up @@ -194,7 +194,7 @@ async def asyncio_detailed(
body: UpdateMemoryOp,
idempotency_key: None | str | Unset = UNSET,
) -> Response[ErrorResponse | OperationEnvelope]:
"""Update Memory (edit a semantic memory)
"""Update a Semantic Memory

Partially update a stored memory by its server-generated id. Only supplied fields change; the memory
is re-embedded when `text` changes.
Expand Down Expand Up @@ -237,7 +237,7 @@ async def asyncio(
body: UpdateMemoryOp,
idempotency_key: None | str | Unset = UNSET,
) -> ErrorResponse | OperationEnvelope | None:
"""Update Memory (edit a semantic memory)
"""Update a Semantic Memory

Partially update a stored memory by its server-generated id. Only supplied fields change; the memory
is re-embedded when `text` changes.
Expand Down
4 changes: 2 additions & 2 deletions robosystems_client/clients/document_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,10 +9,10 @@
from pathlib import Path
from typing import Any, Dict, List, Optional

from ..api.content_operations.op_delete_document import (
from ..api.content_operations.delete_document import (
sync_detailed as delete_document,
)
from ..api.content_operations.op_index_document import (
from ..api.content_operations.index_document import (
sync_detailed as index_document,
)
from ..api.documents.get_document import sync_detailed as get_document
Expand Down
6 changes: 3 additions & 3 deletions robosystems_client/clients/file_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,10 +11,10 @@
import logging
import httpx

from ..api.content_operations.op_create_file_upload import (
from ..api.content_operations.create_file_upload import (
sync_detailed as create_file_upload,
)
from ..api.content_operations.op_ingest_file import (
from ..api.content_operations.ingest_file import (
sync_detailed as ingest_file,
)
from ..api.files.list_files import (
Expand All @@ -23,7 +23,7 @@
from ..api.files.get_file import (
sync_detailed as get_file,
)
from ..api.content_operations.op_delete_file import (
from ..api.content_operations.delete_file import (
sync_detailed as delete_file,
)
from ..models.file_upload_request import FileUploadRequest
Expand Down
6 changes: 3 additions & 3 deletions robosystems_client/clients/graph_client.py
Original file line number Diff line number Diff line change
Expand Up @@ -322,7 +322,7 @@ def materialize(
Returns:
MaterializationResult with detailed execution information
"""
from ..api.graph_operations.op_materialize import (
from ..api.graph_operations.materialize import (
sync_detailed as materialize_graph,
)

Expand Down Expand Up @@ -469,7 +469,7 @@ def change_tier(
RuntimeError: If the tier change fails or the operation errors out
TimeoutError: If the operation does not complete within timeout
"""
from ..api.graph_operations.op_change_tier import sync_detailed as change_tier_op
from ..api.graph_operations.change_tier import sync_detailed as change_tier_op
from ..models.change_tier_op import ChangeTierOp

client = self._get_authenticated_client()
Expand Down Expand Up @@ -709,7 +709,7 @@ def delete_graph(
Raises:
RuntimeError: If the deletion request fails.
"""
from ..api.graph_operations.op_delete_graph import (
from ..api.graph_operations.delete_graph import (
sync_detailed as op_delete_graph,
)
from ..models.delete_graph_op import DeleteGraphOp
Expand Down
Loading
Loading