Skip to content

Commit 1f92f67

Browse files
authored
Delegate api key instantiation to Axiomatic client
1 parent b2e5b94 commit 1f92f67

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

src/axiomatic/magic.py

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -13,18 +13,16 @@ class AXMagic:
1313
"""Class implementing magic functions for IPython.
1414
Import with `%load_ext axiomatic.magic`."""
1515

16-
api_key: str
1716
client: Axiomatic
1817
query: str
1918

20-
def __init__(self, api_key: Optional[str] = os.getenv("AXIOMATIC_API_KEY")):
19+
def __init__(self):
2120
self.folder = platformdirs.user_config_dir("axiomatic")
2221

23-
self.api_key = api_key
24-
self.client = Axiomatic(api_key=api_key)
22+
self.client = Axiomatic()
2523
self.query = ""
2624

27-
def ax_api(self, query):
25+
def ax_api(self, *args, **kwargs):
2826
pass
2927

3028
def axquery(self, query, cell=None):

0 commit comments

Comments
 (0)