Skip to content

Commit

Permalink
feat: Updated agents-api/agents_api/models/docs/ge
Browse files Browse the repository at this point in the history
  • Loading branch information
sweep-ai[bot] committed Apr 13, 2024
1 parent 5e53710 commit 9bc83bd
Showing 1 changed file with 13 additions and 0 deletions.
13 changes: 13 additions & 0 deletions agents-api/agents_api/models/docs/get_docs.py
Original file line number Diff line number Diff line change
@@ -1,3 +1,5 @@
"""Module for retrieving document snippets from the CozoDB based on document IDs."""

from typing import Literal
from uuid import UUID

Expand All @@ -7,6 +9,17 @@
from ...clients.cozo import client


"""
Retrieves snippets of documents by their ID from the CozoDB.
Parameters:
owner_type (Literal["user", "agent"]): The type of the owner of the document.
doc_id (UUID): The unique identifier of the document.
client (CozoClient, optional): The CozoDB client instance. Defaults to a pre-configured client.
Returns:
pd.DataFrame: A DataFrame containing the document snippets and related metadata.
"""
def get_docs_snippets_by_id_query(
owner_type: Literal["user", "agent"],
doc_id: UUID,
Expand Down

0 comments on commit 9bc83bd

Please sign in to comment.