relaxai 0.5.1__py3-none-any.whl → 0.5.2__py3-none-any.whl

relaxai/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "relaxai"
-__version__ = "0.5.1"  # x-release-please-version
+__version__ = "0.5.2"  # x-release-please-version

relaxai/resources/deep_research.py

@@ -7,7 +7,7 @@ from typing import Iterable
 import httpx
 
 from ..types import deep_research_create_params
-from .._types import Body, Omit, Query, Headers, NoneType, NotGiven, SequenceNotStr, omit, not_given
+from .._types import Body, Omit, Query, Headers, NotGiven, SequenceNotStr, omit, not_given
 from .._utils import maybe_transform, async_maybe_transform
 from .._compat import cached_property
 from .._resource import SyncAPIResource, AsyncAPIResource
@@ -19,6 +19,7 @@ from .._response import (
 )
 from .._base_client import make_request_options
 from ..types.stream_options_param import StreamOptionsParam
+from ..types.deepresearch_response import DeepresearchResponse
 from ..types.chat_completion_message_param import ChatCompletionMessageParam
 
 __all__ = ["DeepResearchResource", "AsyncDeepResearchResource"]
@@ -64,7 +65,7 @@ class DeepResearchResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> None:
+    ) -> DeepresearchResponse:
         """
         Performs deep research on a given topic and returns a detailed report.
 
@@ -77,7 +78,6 @@ class DeepResearchResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
         return self._post(
             "/v1/deep-research",
             body=maybe_transform(
@@ -99,7 +99,7 @@ class DeepResearchResource(SyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=NoneType,
+            cast_to=DeepresearchResponse,
         )
 
 
@@ -143,7 +143,7 @@ class AsyncDeepResearchResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> None:
+    ) -> DeepresearchResponse:
         """
         Performs deep research on a given topic and returns a detailed report.
 
@@ -156,7 +156,6 @@ class AsyncDeepResearchResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
         return await self._post(
             "/v1/deep-research",
             body=await async_maybe_transform(
@@ -178,7 +177,7 @@ class AsyncDeepResearchResource(AsyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=NoneType,
+            cast_to=DeepresearchResponse,
         )
 
 

relaxai/resources/tools.py

@@ -5,7 +5,7 @@ from __future__ import annotations
 import httpx
 
 from ..types import tool_execute_code_params
-from .._types import Body, Omit, Query, Headers, NoneType, NotGiven, SequenceNotStr, omit, not_given
+from .._types import Body, Omit, Query, Headers, NotGiven, SequenceNotStr, omit, not_given
 from .._utils import maybe_transform, async_maybe_transform
 from .._compat import cached_property
 from .._resource import SyncAPIResource, AsyncAPIResource
@@ -16,6 +16,7 @@ from .._response import (
     async_to_streamed_response_wrapper,
 )
 from .._base_client import make_request_options
+from ..types.tool_response import ToolResponse
 
 __all__ = ["ToolsResource", "AsyncToolsResource"]
 
@@ -52,7 +53,7 @@ class ToolsResource(SyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> None:
+    ) -> ToolResponse:
         """
         Run the given code in a secure environment and return the output.
 
@@ -65,7 +66,6 @@ class ToolsResource(SyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
         return self._post(
             "/v1/tools/code",
             body=maybe_transform(
@@ -79,7 +79,7 @@ class ToolsResource(SyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=NoneType,
+            cast_to=ToolResponse,
         )
 
 
@@ -115,7 +115,7 @@ class AsyncToolsResource(AsyncAPIResource):
         extra_query: Query | None = None,
         extra_body: Body | None = None,
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
-    ) -> None:
+    ) -> ToolResponse:
         """
         Run the given code in a secure environment and return the output.
 
@@ -128,7 +128,6 @@ class AsyncToolsResource(AsyncAPIResource):
 
           timeout: Override the client-level default timeout for this request, in seconds
         """
-        extra_headers = {"Accept": "*/*", **(extra_headers or {})}
         return await self._post(
             "/v1/tools/code",
             body=await async_maybe_transform(
@@ -142,7 +141,7 @@ class AsyncToolsResource(AsyncAPIResource):
             options=make_request_options(
                 extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
             ),
-            cast_to=NoneType,
+            cast_to=ToolResponse,
         )
 
 

relaxai/types/__init__.py

@@ -10,10 +10,12 @@ from .shared import (
 )
 from .model_list import ModelList as ModelList
 from .function_call import FunctionCall as FunctionCall
+from .tool_response import ToolResponse as ToolResponse
 from .health_response import HealthResponse as HealthResponse
 from .embedding_response import EmbeddingResponse as EmbeddingResponse
 from .function_call_param import FunctionCallParam as FunctionCallParam
 from .stream_options_param import StreamOptionsParam as StreamOptionsParam
+from .deepresearch_response import DeepresearchResponse as DeepresearchResponse
 from .content_filter_results import ContentFilterResults as ContentFilterResults
 from .chat_completion_message import ChatCompletionMessage as ChatCompletionMessage
 from .chat_completion_response import ChatCompletionResponse as ChatCompletionResponse

relaxai/types/deepresearch_response.py

@@ -0,0 +1,115 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List, Optional
+
+from .._models import BaseModel
+from .function_call import FunctionCall
+from .shared.openai_usage import OpenAIUsage
+
+__all__ = [
+    "DeepresearchResponse",
+    "Choice",
+    "ChoiceMessage",
+    "ChoiceMessageAnnotation",
+    "ChoiceMessageAnnotationURLCitation",
+    "ChoiceMessageToolCall",
+    "ChoiceLogprobs",
+    "ChoiceLogprobsContent",
+    "ChoiceLogprobsContentTopLogprob",
+]
+
+
+class ChoiceMessageAnnotationURLCitation(BaseModel):
+    end_index: int
+
+    start_index: int
+
+    title: str
+
+    url: str
+
+
+class ChoiceMessageAnnotation(BaseModel):
+    type: str
+
+    url_citation: Optional[ChoiceMessageAnnotationURLCitation] = None
+
+
+class ChoiceMessageToolCall(BaseModel):
+    function: FunctionCall
+
+    type: str
+
+    id: Optional[str] = None
+
+    index: Optional[int] = None
+
+
+class ChoiceMessage(BaseModel):
+    role: str
+
+    annotations: Optional[List[ChoiceMessageAnnotation]] = None
+
+    content: Optional[str] = None
+
+    function_call: Optional[FunctionCall] = None
+
+    name: Optional[str] = None
+
+    reasoning_content: Optional[str] = None
+
+    refusal: Optional[str] = None
+
+    tool_call_id: Optional[str] = None
+
+    tool_calls: Optional[List[ChoiceMessageToolCall]] = None
+
+
+class ChoiceLogprobsContentTopLogprob(BaseModel):
+    token: str
+
+    logprob: float
+
+    bytes: Optional[str] = None
+
+
+class ChoiceLogprobsContent(BaseModel):
+    token: str
+
+    logprob: float
+
+    top_logprobs: List[ChoiceLogprobsContentTopLogprob]
+
+    bytes: Optional[str] = None
+
+
+class ChoiceLogprobs(BaseModel):
+    content: List[ChoiceLogprobsContent]
+
+
+class Choice(BaseModel):
+    finish_reason: str
+
+    index: int
+
+    message: ChoiceMessage
+
+    logprobs: Optional[ChoiceLogprobs] = None
+
+
+class DeepresearchResponse(BaseModel):
+    id: str
+
+    choices: List[Choice]
+
+    created: int
+
+    model: str
+
+    object: str
+
+    system_fingerprint: str
+
+    usage: OpenAIUsage
+
+    service_tier: Optional[str] = None

relaxai/types/tool_response.py

@@ -0,0 +1,33 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from typing import List, Optional
+
+from .._models import BaseModel
+
+__all__ = ["ToolResponse", "Plot"]
+
+
+class Plot(BaseModel):
+    content_base64: str
+
+    filename: str
+
+    format: str
+
+
+class ToolResponse(BaseModel):
+    execution_time: float
+
+    exit_code: int
+
+    security_checked: bool
+
+    stderr: str
+
+    stdout: str
+
+    success: bool
+
+    plots: Optional[List[Plot]] = None
+
+    violations: Optional[List[str]] = None

{relaxai-0.5.1.dist-info → relaxai-0.5.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: relaxai
-Version: 0.5.1
+Version: 0.5.2
 Summary: The official Python library for the relaxai API
 Project-URL: Homepage, https://github.com/relax-ai/python-sdk
 Project-URL: Repository, https://github.com/relax-ai/python-sdk

{relaxai-0.5.1.dist-info → relaxai-0.5.2.dist-info}/RECORD

@@ -11,7 +11,7 @@ relaxai/_resource.py,sha256=v7qxjafEaUaEfecj54WhuhW2hupVzM8Os0EIHVRmbkc,1106
 relaxai/_response.py,sha256=URFVS4ivxSoLsvosTN0Rc3f5vbBBb0DIB4Eh2XPGa_Q,28794
 relaxai/_streaming.py,sha256=Nr4O_q1hh35alNVqo46KKf9ZTbRp-IoXjGFCCNJC1fA,10104
 relaxai/_types.py,sha256=v5coMcqP3kIeECx9ks-aaxiodh2l0UEKv6UI4creii4,7237
-relaxai/_version.py,sha256=1hJwi-xdzLkl0tFMXzd2ti9Q6rWWU_nsJ2aShj75jbg,159
+relaxai/_version.py,sha256=svd-bh0pKXXETQobMzg46j5yTdeXt2xCe_J206pc2Tg,159
 relaxai/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 relaxai/_utils/__init__.py,sha256=7fch0GT9zpNnErbciSpUNa-SjTxxjY6kxHxKMOM4AGs,2305
 relaxai/_utils/_compat.py,sha256=D8gtAvjJQrDWt9upS0XaG9Rr5l1QhiAx_I_1utT_tt0,1195
@@ -28,17 +28,18 @@ relaxai/_utils/_utils.py,sha256=0dDqauUbVZEXV0NVl7Bwu904Wwo5eyFCZpQThhFNhyA,1225
 relaxai/lib/.keep,sha256=wuNrz-5SXo3jJaJOJgz4vFHM41YH_g20F5cRQo0vLes,224
 relaxai/resources/__init__.py,sha256=PN55gCB7h9TokDnEC4NT3jPj17828geZDkMh6wv9uKE,2444
 relaxai/resources/chat.py,sha256=yeDuAAJk3dW1ysw7nvO2euzPzcjeHAxT-Y-9BI5l40Y,13053
-relaxai/resources/deep_research.py,sha256=n7dzP9efBLKThZOW9yF4cCF2_qOGc3IzcwjYG21th3g,8499
+relaxai/resources/deep_research.py,sha256=87098Z1XBBd52foI-31x7pYU09BenVD0VoM5lIlRyc8,8474
 relaxai/resources/embeddings.py,sha256=BpG5jmG2O-hVnRawd834JQLvKeC7rzvfaOAyxXapGbg,6998
 relaxai/resources/models.py,sha256=Zkp7vxtB5xp2vDHOsWB2XQhcj-52AAusvFQAiS2aVWE,7856
-relaxai/resources/tools.py,sha256=wqz_QBd3rI0jbPL8XhVwSP3hEIJzmT9-lFv8Kgr3eO4,6484
-relaxai/types/__init__.py,sha256=nnoOEpJGoxrxdR1FDMiAkXyhEVI0c8H003F8ppCecR4,1557
+relaxai/resources/tools.py,sha256=n6iE8e-KtqGJ-YaiRHdYUTlVzi9AOc4vd9ioP7jE5go,6411
+relaxai/types/__init__.py,sha256=TFMRnoRboP2fVVpQpbKUrzgHbjd1G0JKJULvEyR_s7o,1693
 relaxai/types/chat_completion_message.py,sha256=t-dKQxWfqUr8UsX5Fs3NpN9YfRJ5oJvozfZtWn3JF8M,1544
 relaxai/types/chat_completion_message_param.py,sha256=X4oAp6FQwue-SEibwr3KcxhJNQZiaiBhz2Q9mMHY5_M,1485
 relaxai/types/chat_completion_response.py,sha256=II4PjVfW60aa9aX5mApsseDerQVieTlt9BLim2fY_RE,1564
 relaxai/types/chat_create_completion_params.py,sha256=q8c31zNDMfjYr5Ai0k-EfDXqRTfXyLGbf2DiuNgZtN8,3055
 relaxai/types/content_filter_results.py,sha256=vOaHu4jdGKVX4vqqCZohTYDsKDXlvfh57Ozb_95-c2U,999
 relaxai/types/deep_research_create_params.py,sha256=T0X5-8ck0_no5fHWQGM2GoSjJ1ntYDHlBXl_odGFK9Y,1239
+relaxai/types/deepresearch_response.py,sha256=QO2wOxFDzd5eVTdp9scVEwG8-g7vYIPBg4g_dSj-5Ww,2051
 relaxai/types/embedding_create_embedding_params.py,sha256=76aUOwC4LMjfMesJTjrjUh_FzVkDcRe_2rRzMTcdaag,399
 relaxai/types/embedding_response.py,sha256=N7SS9DhAbG5kEWF6RxYc8gl_rUvY9c1mSoT8v4vwahA,549
 relaxai/types/function_call.py,sha256=n_agCO6Vrh5YZH1SZ6GWOV4mk2IFPeHJA_WkZrPGPMw,277
@@ -49,11 +50,12 @@ relaxai/types/model.py,sha256=Du9lb9dn846N1fPrKtDsPSLczkqnzxVAJPb8GMqyVJ8,764
 relaxai/types/model_list.py,sha256=9lvzYRTBYTVudP9_28Ai0lUfc82eXOmz0qZ_rwCzxIU,364
 relaxai/types/stream_options_param.py,sha256=JEom-Fs3ORSTD86NTan-9ZjlKYfPEvSYXViUhOXPCb0,273
 relaxai/types/tool_execute_code_params.py,sha256=pHhBZiOQ5e7bYe0kXAfyKld3rpFgERAt1zm2iURWZBc,387
+relaxai/types/tool_response.py,sha256=OMHBB4wq34OMpqWzemRMRanV1oyRy1UcF6prRVrj-HM,519
 relaxai/types/shared/__init__.py,sha256=rVHWDutDMA_BYQSHahIzfwcaDAwE3zyBvTiPRI1lSFo,346
 relaxai/types/shared/openai_completion_tokens_details.py,sha256=kjDBsTKvzhw-8UA3t7CQ6qvAo4Lmnup5VEFU2QzBynw,338
 relaxai/types/shared/openai_prompt_tokens_details.py,sha256=wdPCsobx-VBw3MgfSCY4HXrGnMtw6YXiepKpXw1kZAE,253
 relaxai/types/shared/openai_usage.py,sha256=W-mht7UXuWne_eFXCBXHpf2HEsjzfXo2jPXOdKNEdbQ,513
-relaxai-0.5.1.dist-info/METADATA,sha256=3oTc09bp3bgJOd7ZFF3PGKb-V97-no5rjyDXxXdrK5Y,15304
-relaxai-0.5.1.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-relaxai-0.5.1.dist-info/licenses/LICENSE,sha256=B0u5zBqmaNCfIZooH665f2J7yFqMmSjQxZtnJhCuID8,11337
-relaxai-0.5.1.dist-info/RECORD,,
+relaxai-0.5.2.dist-info/METADATA,sha256=krJT1itRLyDNqbI_3q2UHk6JcyuZQUWMkS5D8OmCuG4,15304
+relaxai-0.5.2.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+relaxai-0.5.2.dist-info/licenses/LICENSE,sha256=B0u5zBqmaNCfIZooH665f2J7yFqMmSjQxZtnJhCuID8,11337
+relaxai-0.5.2.dist-info/RECORD,,