Release 0.0.3

Author: fern-api[bot]

README.md

@@ -24,7 +24,6 @@ from axiomatic import Axiomatic
 
 client = Axiomatic(
     api_key="YOUR_API_KEY",
-    base_url="https://yourhost.com/path/to/api",
 )
 client.pic.extract()
 ```
@@ -40,7 +39,6 @@ from axiomatic import AsyncAxiomatic
 
 client = AsyncAxiomatic(
     api_key="YOUR_API_KEY",
-    base_url="https://yourhost.com/path/to/api",
 )
 
 

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "axiomatic"
-version = "0.0.2"
+version = "0.0.3"
 description = ""
 readme = "README.md"
 authors = []

reference.md

@@ -16,7 +16,6 @@ from axiomatic import Axiomatic
 
 client = Axiomatic(
     api_key="YOUR_API_KEY",
-    base_url="https://yourhost.com/path/to/api",
 )
 client.health_check_health_check_get()
 
@@ -78,7 +77,6 @@ from axiomatic import Axiomatic
 
 client = Axiomatic(
     api_key="YOUR_API_KEY",
-    base_url="https://yourhost.com/path/to/api",
 )
 client.pic.extract()
 
@@ -149,7 +147,6 @@ from axiomatic import Axiomatic
 
 client = Axiomatic(
     api_key="YOUR_API_KEY",
-    base_url="https://yourhost.com/path/to/api",
 )
 client.pic.extract_from_text(
     query="query",
@@ -220,7 +217,6 @@ from axiomatic import Axiomatic
 
 client = Axiomatic(
     api_key="YOUR_API_KEY",
-    base_url="https://yourhost.com/path/to/api",
 )
 client.pic.extract_from_paper()
 
@@ -291,7 +287,6 @@ from axiomatic import Axiomatic, StatementInput
 
 client = Axiomatic(
     api_key="YOUR_API_KEY",
-    base_url="https://yourhost.com/path/to/api",
 )
 client.pic.identify(
     statements=[
@@ -367,7 +362,6 @@ from axiomatic import Axiomatic
 
 client = Axiomatic(
     api_key="YOUR_API_KEY",
-    base_url="https://yourhost.com/path/to/api",
 )
 client.pic.identify_with_context(
     statements="statements",
@@ -446,7 +440,6 @@ from axiomatic import Axiomatic
 
 client = Axiomatic(
     api_key="YOUR_API_KEY",
-    base_url="https://yourhost.com/path/to/api",
 )
 client.pic.synthesize_circuit_from_paper()
 
@@ -517,7 +510,6 @@ from axiomatic import Axiomatic
 
 client = Axiomatic(
     api_key="YOUR_API_KEY",
-    base_url="https://yourhost.com/path/to/api",
 )
 client.pic.synthesize_circuit_from_text(
     query="query",
@@ -575,7 +567,6 @@ from axiomatic import Axiomatic
 
 client = Axiomatic(
     api_key="YOUR_API_KEY",
-    base_url="https://yourhost.com/path/to/api",
 )
 client.lean.lean_execute()
 
@@ -646,7 +637,6 @@ from axiomatic import Axiomatic
 
 client = Axiomatic(
     api_key="YOUR_API_KEY",
-    base_url="https://yourhost.com/path/to/api",
 )
 client.lean.suggest(
     prompt="prompt",
@@ -712,7 +702,6 @@ from axiomatic import Axiomatic
 
 client = Axiomatic(
     api_key="YOUR_API_KEY",
-    base_url="https://yourhost.com/path/to/api",
 )
 client.lean.z_3_execute(
     code="code",
@@ -784,7 +773,6 @@ from axiomatic import Axiomatic
 
 client = Axiomatic(
     api_key="YOUR_API_KEY",
-    base_url="https://yourhost.com/path/to/api",
 )
 client.experimental.assistant(
     query="query",
@@ -864,7 +852,6 @@ from axiomatic import Axiomatic
 
 client = Axiomatic(
     api_key="YOUR_API_KEY",
-    base_url="https://yourhost.com/path/to/api",
 )
 client.experimental.synthesize(
     query="query",
@@ -922,7 +909,6 @@ from axiomatic import Axiomatic
 
 client = Axiomatic(
     api_key="YOUR_API_KEY",
-    base_url="https://yourhost.com/path/to/api",
 )
 client.generic.extract(
     query="query",

src/axiomatic/__init__.py

@@ -20,11 +20,13 @@
 from .errors import UnprocessableEntityError
 from . import experimental, generic, lean, pic
 from .client import AsyncAxiomatic, Axiomatic
+from .environment import AxiomaticEnvironment
 from .version import __version__
 
 __all__ = [
     "AsyncAxiomatic",
     "Axiomatic",
+    "AxiomaticEnvironment",
     "CircuitImage",
     "CodeSynthesisResponse",
     "Edge",

src/axiomatic/client.py

@@ -1,6 +1,7 @@
 # This file was auto-generated by Fern from our API Definition.
 
 import typing
+from .environment import AxiomaticEnvironment
 import httpx
 from .core.client_wrapper import SyncClientWrapper
 from .pic.client import PicClient
@@ -24,9 +25,18 @@ class Axiomatic:
 
     Parameters
     ----------
-    base_url : str
+    base_url : typing.Optional[str]
         The base url to use for requests from the client.
 
+    environment : AxiomaticEnvironment
+        The environment to use for requests from the client. from .environment import AxiomaticEnvironment
+
+
+
+        Defaults to AxiomaticEnvironment.DEFAULT
+
+
+
     api_key : str
     timeout : typing.Optional[float]
         The timeout to be used, in seconds, for requests. By default the timeout is 60 seconds, unless a custom httpx client is used, in which case this default is not enforced.
@@ -43,22 +53,22 @@ class Axiomatic:
 
     client = Axiomatic(
         api_key="YOUR_API_KEY",
-        base_url="https://yourhost.com/path/to/api",
     )
     """
 
     def __init__(
         self,
         *,
-        base_url: str,
+        base_url: typing.Optional[str] = None,
+        environment: AxiomaticEnvironment = AxiomaticEnvironment.DEFAULT,
         api_key: str,
         timeout: typing.Optional[float] = None,
         follow_redirects: typing.Optional[bool] = True,
         httpx_client: typing.Optional[httpx.Client] = None,
     ):
         _defaulted_timeout = timeout if timeout is not None else 60 if httpx_client is None else None
         self._client_wrapper = SyncClientWrapper(
-            base_url=base_url,
+            base_url=_get_base_url(base_url=base_url, environment=environment),
             api_key=api_key,
             httpx_client=httpx_client
             if httpx_client is not None
@@ -92,7 +102,6 @@ def health_check_health_check_get(
 
         client = Axiomatic(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
         client.health_check_health_check_get()
         """
@@ -122,9 +131,18 @@ class AsyncAxiomatic:
 
     Parameters
     ----------
-    base_url : str
+    base_url : typing.Optional[str]
         The base url to use for requests from the client.
 
+    environment : AxiomaticEnvironment
+        The environment to use for requests from the client. from .environment import AxiomaticEnvironment
+
+
+
+        Defaults to AxiomaticEnvironment.DEFAULT
+
+
+
     api_key : str
     timeout : typing.Optional[float]
         The timeout to be used, in seconds, for requests. By default the timeout is 60 seconds, unless a custom httpx client is used, in which case this default is not enforced.
@@ -141,22 +159,22 @@ class AsyncAxiomatic:
 
     client = AsyncAxiomatic(
         api_key="YOUR_API_KEY",
-        base_url="https://yourhost.com/path/to/api",
     )
     """
 
     def __init__(
         self,
         *,
-        base_url: str,
+        base_url: typing.Optional[str] = None,
+        environment: AxiomaticEnvironment = AxiomaticEnvironment.DEFAULT,
         api_key: str,
         timeout: typing.Optional[float] = None,
         follow_redirects: typing.Optional[bool] = True,
         httpx_client: typing.Optional[httpx.AsyncClient] = None,
     ):
         _defaulted_timeout = timeout if timeout is not None else 60 if httpx_client is None else None
         self._client_wrapper = AsyncClientWrapper(
-            base_url=base_url,
+            base_url=_get_base_url(base_url=base_url, environment=environment),
             api_key=api_key,
             httpx_client=httpx_client
             if httpx_client is not None
@@ -192,7 +210,6 @@ async def health_check_health_check_get(
 
         client = AsyncAxiomatic(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
 
 
@@ -220,3 +237,12 @@ async def main() -> None:
         except JSONDecodeError:
             raise ApiError(status_code=_response.status_code, body=_response.text)
         raise ApiError(status_code=_response.status_code, body=_response_json)
+
+
+def _get_base_url(*, base_url: typing.Optional[str] = None, environment: AxiomaticEnvironment) -> str:
+    if base_url is not None:
+        return base_url
+    elif environment is not None:
+        return environment.value
+    else:
+        raise Exception("Please pass in either base_url or environment to construct the client")

src/axiomatic/core/client_wrapper.py

@@ -16,7 +16,7 @@ def get_headers(self) -> typing.Dict[str, str]:
         headers: typing.Dict[str, str] = {
             "X-Fern-Language": "Python",
             "X-Fern-SDK-Name": "axiomatic",
-            "X-Fern-SDK-Version": "0.0.2",
+            "X-Fern-SDK-Version": "0.0.3",
         }
         headers["x-api-key"] = self.api_key
         return headers

src/axiomatic/environment.py

@@ -0,0 +1,7 @@
+# This file was auto-generated by Fern from our API Definition.
+
+import enum
+
+
+class AxiomaticEnvironment(enum.Enum):
+    DEFAULT = "https://api.axiomatic-ai.com"

src/axiomatic/experimental/client.py

@@ -43,7 +43,6 @@ def assistant(
 
         client = Axiomatic(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
         client.experimental.assistant(
             query="query",
@@ -107,7 +106,6 @@ def synthesize(
 
         client = Axiomatic(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
         client.experimental.synthesize(
             query="query",
@@ -178,7 +176,6 @@ async def assistant(
 
         client = AsyncAxiomatic(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
 
 
@@ -250,7 +247,6 @@ async def synthesize(
 
         client = AsyncAxiomatic(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
 
 

src/axiomatic/generic/client.py

@@ -40,7 +40,6 @@ def extract(
 
         client = Axiomatic(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
         client.generic.extract(
             query="query",
@@ -111,7 +110,6 @@ async def extract(
 
         client = AsyncAxiomatic(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
 
 

src/axiomatic/lean/client.py

@@ -49,7 +49,6 @@ def lean_execute(
 
         client = Axiomatic(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
         client.lean.lean_execute()
         """
@@ -115,7 +114,6 @@ def suggest(
 
         client = Axiomatic(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
         client.lean.suggest(
             prompt="prompt",
@@ -181,7 +179,6 @@ def z_3_execute(
 
         client = Axiomatic(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
         client.lean.z_3_execute(
             code="code",
@@ -261,7 +258,6 @@ async def lean_execute(
 
         client = AsyncAxiomatic(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
 
 
@@ -335,7 +331,6 @@ async def suggest(
 
         client = AsyncAxiomatic(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )
 
 
@@ -409,7 +404,6 @@ async def z_3_execute(
 
         client = AsyncAxiomatic(
             api_key="YOUR_API_KEY",
-            base_url="https://yourhost.com/path/to/api",
         )