athena-intelligence 0.1.36__tar.gz → 0.1.38__tar.gz

{athena_intelligence-0.1.36 → athena_intelligence-0.1.38}/PKG-INFO

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

{athena_intelligence-0.1.36 → athena_intelligence-0.1.38}/pyproject.toml

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

{athena_intelligence-0.1.36 → athena_intelligence-0.1.38}/src/athena/__init__.py

@@ -9,6 +9,7 @@ from .types import (
     MessageOutDto,
     Model,
     Report,
+    ScrapeMarkdownResult,
     Snippet,
     SqlResults,
     StatusEnum,
@@ -31,6 +32,7 @@ __all__ = [
     "MessageOutDto",
     "Model",
     "Report",
+    "ScrapeMarkdownResult",
     "Snippet",
     "SqlResults",
     "StatusEnum",

{athena_intelligence-0.1.36 → athena_intelligence-0.1.38}/src/athena/core/client_wrapper.py

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

{athena_intelligence-0.1.36 → athena_intelligence-0.1.38}/src/athena/search/client.py

@@ -11,6 +11,7 @@ from ..core.remove_none_from_dict import remove_none_from_dict
 from ..core.request_options import RequestOptions
 from ..errors.unprocessable_entity_error import UnprocessableEntityError
 from ..types.http_validation_error import HttpValidationError
+from ..types.scrape_markdown_result import ScrapeMarkdownResult
 from ..types.url_result import UrlResult
 
 try:
@@ -107,7 +108,9 @@ class SearchClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    def get_markdown(self, *, url: str, request_options: typing.Optional[RequestOptions] = None) -> UrlResult:
+    def get_markdown(
+        self, *, url: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> ScrapeMarkdownResult:
         """
         Parameters:
             - url: str.
@@ -150,7 +153,7 @@ class SearchClient:
             max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(UrlResult, _response.json())  # type: ignore
+            return pydantic.parse_obj_as(ScrapeMarkdownResult, _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:
@@ -245,7 +248,9 @@ class AsyncSearchClient:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
 
-    async def get_markdown(self, *, url: str, request_options: typing.Optional[RequestOptions] = None) -> UrlResult:
+    async def get_markdown(
+        self, *, url: str, request_options: typing.Optional[RequestOptions] = None
+    ) -> ScrapeMarkdownResult:
         """
         Parameters:
             - url: str.
@@ -288,7 +293,7 @@ class AsyncSearchClient:
             max_retries=request_options.get("max_retries") if request_options is not None else 0,  # type: ignore
         )
         if 200 <= _response.status_code < 300:
-            return pydantic.parse_obj_as(UrlResult, _response.json())  # type: ignore
+            return pydantic.parse_obj_as(ScrapeMarkdownResult, _response.json())  # type: ignore
         if _response.status_code == 422:
             raise UnprocessableEntityError(pydantic.parse_obj_as(HttpValidationError, _response.json()))  # type: ignore
         try:

{athena_intelligence-0.1.36 → athena_intelligence-0.1.38}/src/athena/types/__init__.py

@@ -8,6 +8,7 @@ from .message_out import MessageOut
 from .message_out_dto import MessageOutDto
 from .model import Model
 from .report import Report
+from .scrape_markdown_result import ScrapeMarkdownResult
 from .snippet import Snippet
 from .sql_results import SqlResults
 from .status_enum import StatusEnum
@@ -25,6 +26,7 @@ __all__ = [
     "MessageOutDto",
     "Model",
     "Report",
+    "ScrapeMarkdownResult",
     "Snippet",
     "SqlResults",
     "StatusEnum",

athena_intelligence-0.1.38/src/athena/types/scrape_markdown_result.py

@@ -0,0 +1,28 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import datetime as dt
+import typing
+
+from ..core.datetime_utils import serialize_datetime
+
+try:
+    import pydantic.v1 as pydantic  # type: ignore
+except ImportError:
+    import pydantic  # type: ignore
+
+
+class ScrapeMarkdownResult(pydantic.BaseModel):
+    markdown: str
+
+    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
+        json_encoders = {dt.datetime: serialize_datetime}