athena-intelligence 0.1.61__tar.gz → 0.1.63__tar.gz

{athena_intelligence-0.1.61 → athena_intelligence-0.1.63}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: athena-intelligence
-Version: 0.1.61
+Version: 0.1.63
 Summary: 
 Requires-Python: >=3.8,<4.0
 Classifier: Programming Language :: Python :: 3

{athena_intelligence-0.1.61 → athena_intelligence-0.1.63}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "athena-intelligence"
-version = "0.1.61"
+version = "0.1.63"
 description = ""
 readme = "README.md"
 authors = []

{athena_intelligence-0.1.61 → athena_intelligence-0.1.63}/src/athena/__init__.py

@@ -4,6 +4,7 @@ from .types import (
     Dataset,
     Document,
     ExcecuteToolFirstWorkflowOut,
+    FileDataResponse,
     FirecrawlScrapeUrlDataReponseDto,
     FirecrawlScrapeUrlMetadata,
     GetDatasetsResponse,
@@ -16,6 +17,7 @@ from .types import (
     MessageOut,
     MessageOutDto,
     Model,
+    PublishFormats,
     Report,
     ResearcherOut,
     Snippet,
@@ -38,6 +40,7 @@ __all__ = [
     "Dataset",
     "Document",
     "ExcecuteToolFirstWorkflowOut",
+    "FileDataResponse",
     "FirecrawlScrapeUrlDataReponseDto",
     "FirecrawlScrapeUrlMetadata",
     "GetDatasetsResponse",
@@ -50,6 +53,7 @@ __all__ = [
     "MessageOut",
     "MessageOutDto",
     "Model",
+    "PublishFormats",
     "Report",
     "ResearcherOut",
     "Snippet",

{athena_intelligence-0.1.61 → athena_intelligence-0.1.63}/src/athena/core/client_wrapper.py

@@ -17,7 +17,7 @@ class BaseClientWrapper:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "athena-intelligence",
-            "X-Fern-SDK-Version": "0.1.61",
+            "X-Fern-SDK-Version": "0.1.63",
         }
         headers["X-API-KEY"] = self.api_key
         return headers

{athena_intelligence-0.1.61 → athena_intelligence-0.1.63}/src/athena/tools/client.py

@@ -16,6 +16,7 @@ from ..types.firecrawl_scrape_url_data_reponse_dto import FirecrawlScrapeUrlData
 from ..types.http_validation_error import HttpValidationError
 from ..types.langchain_documents_request_out import LangchainDocumentsRequestOut
 from ..types.llm_model import LlmModel
+from ..types.publish_formats import PublishFormats
 from ..types.researcher_out import ResearcherOut
 
 # this is used as the default value for optional parameters
@@ -248,6 +249,7 @@ class ToolsClient:
         query: str,
         max_sections: int,
         guidelines: typing.Sequence[str],
+        publish_formats: typing.Optional[PublishFormats] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> ResearcherOut:
         """
@@ -258,8 +260,11 @@ class ToolsClient:
 
             - guidelines: typing.Sequence[str].
 
+            - publish_formats: typing.Optional[PublishFormats].
+
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
+        from athena import PublishFormats
         from athena.client import Athena
 
         client = Athena(
@@ -273,18 +278,30 @@ class ToolsClient:
                 "Each sub section MUST include supporting sources using hyperlinks. If none exist, erase the sub section or rewrite it to be a part of the previous section",
                 "The report MUST be written in english",
             ],
+            publish_formats=PublishFormats(
+                markdown=False,
+                pdf=True,
+                docx=False,
+            ),
         )
         """
+        _request: typing.Dict[str, typing.Any] = {
+            "query": query,
+            "max_sections": max_sections,
+            "guidelines": guidelines,
+        }
+        if publish_formats is not OMIT:
+            _request["publish_formats"] = publish_formats
         _response = self._client_wrapper.httpx_client.request(
             method="POST",
             url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/tools/researcher"),
             params=jsonable_encoder(
                 request_options.get("additional_query_parameters") if request_options is not None else None
             ),
-            json=jsonable_encoder({"query": query, "max_sections": max_sections, "guidelines": guidelines})
+            json=jsonable_encoder(_request)
             if request_options is None or request_options.get("additional_body_parameters") is None
             else {
-                **jsonable_encoder({"query": query, "max_sections": max_sections, "guidelines": guidelines}),
+                **jsonable_encoder(_request),
                 **(jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))),
             },
             headers=jsonable_encoder(
@@ -540,6 +557,7 @@ class AsyncToolsClient:
         query: str,
         max_sections: int,
         guidelines: typing.Sequence[str],
+        publish_formats: typing.Optional[PublishFormats] = OMIT,
         request_options: typing.Optional[RequestOptions] = None,
     ) -> ResearcherOut:
         """
@@ -550,8 +568,11 @@ class AsyncToolsClient:
 
             - guidelines: typing.Sequence[str].
 
+            - publish_formats: typing.Optional[PublishFormats].
+
             - request_options: typing.Optional[RequestOptions]. Request-specific configuration.
         ---
+        from athena import PublishFormats
         from athena.client import AsyncAthena
 
         client = AsyncAthena(
@@ -565,18 +586,30 @@ class AsyncToolsClient:
                 "Each sub section MUST include supporting sources using hyperlinks. If none exist, erase the sub section or rewrite it to be a part of the previous section",
                 "The report MUST be written in english",
             ],
+            publish_formats=PublishFormats(
+                markdown=False,
+                pdf=True,
+                docx=False,
+            ),
         )
         """
+        _request: typing.Dict[str, typing.Any] = {
+            "query": query,
+            "max_sections": max_sections,
+            "guidelines": guidelines,
+        }
+        if publish_formats is not OMIT:
+            _request["publish_formats"] = publish_formats
         _response = await self._client_wrapper.httpx_client.request(
             method="POST",
             url=urllib.parse.urljoin(f"{self._client_wrapper.get_base_url()}/", "api/v0/tools/researcher"),
             params=jsonable_encoder(
                 request_options.get("additional_query_parameters") if request_options is not None else None
             ),
-            json=jsonable_encoder({"query": query, "max_sections": max_sections, "guidelines": guidelines})
+            json=jsonable_encoder(_request)
             if request_options is None or request_options.get("additional_body_parameters") is None
             else {
-                **jsonable_encoder({"query": query, "max_sections": max_sections, "guidelines": guidelines}),
+                **jsonable_encoder(_request),
                 **(jsonable_encoder(remove_none_from_dict(request_options.get("additional_body_parameters", {})))),
             },
             headers=jsonable_encoder(

{athena_intelligence-0.1.61 → athena_intelligence-0.1.63}/src/athena/types/__init__.py

@@ -3,6 +3,7 @@
 from .dataset import Dataset
 from .document import Document
 from .excecute_tool_first_workflow_out import ExcecuteToolFirstWorkflowOut
+from .file_data_response import FileDataResponse
 from .firecrawl_scrape_url_data_reponse_dto import FirecrawlScrapeUrlDataReponseDto
 from .firecrawl_scrape_url_metadata import FirecrawlScrapeUrlMetadata
 from .get_datasets_response import GetDatasetsResponse
@@ -15,6 +16,7 @@ from .map_reduce_chain_out import MapReduceChainOut
 from .message_out import MessageOut
 from .message_out_dto import MessageOutDto
 from .model import Model
+from .publish_formats import PublishFormats
 from .report import Report
 from .researcher_out import ResearcherOut
 from .snippet import Snippet
@@ -31,6 +33,7 @@ __all__ = [
     "Dataset",
     "Document",
     "ExcecuteToolFirstWorkflowOut",
+    "FileDataResponse",
     "FirecrawlScrapeUrlDataReponseDto",
     "FirecrawlScrapeUrlMetadata",
     "GetDatasetsResponse",
@@ -43,6 +46,7 @@ __all__ = [
     "MessageOut",
     "MessageOutDto",
     "Model",
+    "PublishFormats",
     "Report",
     "ResearcherOut",
     "Snippet",

athena_intelligence-0.1.63/src/athena/types/file_data_response.py

@@ -0,0 +1,32 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
+
+
+class FileDataResponse(pydantic_v1.BaseModel):
+    file: typing.Optional[str] = None
+    size: typing.Optional[int] = None
+    filename: typing.Optional[str] = None
+    content_type: typing.Optional[str] = None
+    status: typing.Optional[bool] = None
+    error: typing.Optional[str] = None
+    message: typing.Optional[str] = None
+    athena_doc_id: typing.Optional[str] = None
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

athena_intelligence-0.1.63/src/athena/types/publish_formats.py

@@ -0,0 +1,27 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+from ..core.pydantic_utilities import pydantic_v1
+
+
+class PublishFormats(pydantic_v1.BaseModel):
+    markdown: bool
+    pdf: bool
+    docx: bool
+
+    def json(self, **kwargs: typing.Any) -> str:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().json(**kwargs_with_defaults)
+
+    def dict(self, **kwargs: typing.Any) -> typing.Dict[str, typing.Any]:
+        kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}
+        return super().dict(**kwargs_with_defaults)
+
+    class Config:
+        frozen = True
+        smart_union = True
+        extra = pydantic_v1.Extra.allow
+        json_encoders = {dt.datetime: serialize_datetime}

{athena_intelligence-0.1.61 → athena_intelligence-0.1.63}/src/athena/types/upload_documents_out.py

@@ -5,11 +5,12 @@ import typing
 
 from ..core.datetime_utils import serialize_datetime
 from ..core.pydantic_utilities import pydantic_v1
+from .file_data_response import FileDataResponse
 
 
 class UploadDocumentsOut(pydantic_v1.BaseModel):
     success: bool
-    uploaded_document_names: typing.Optional[typing.List[str]] = None
+    uploaded_documents: typing.Optional[typing.List[FileDataResponse]] = None
 
     def json(self, **kwargs: typing.Any) -> str:
         kwargs_with_defaults: typing.Any = {"by_alias": True, "exclude_unset": True, **kwargs}