open-data-sci 0.1.0__py3-none-any.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (85) hide show
  1. open_data_sci-0.1.0.dist-info/METADATA +629 -0
  2. open_data_sci-0.1.0.dist-info/RECORD +85 -0
  3. open_data_sci-0.1.0.dist-info/WHEEL +4 -0
  4. open_data_sci-0.1.0.dist-info/entry_points.txt +2 -0
  5. open_data_sci-0.1.0.dist-info/licenses/LICENSE +201 -0
  6. opendatasci/__init__.py +47 -0
  7. opendatasci/_tui/__init__.py +1 -0
  8. opendatasci/_tui/adapter.py +102 -0
  9. opendatasci/_tui/app.py +429 -0
  10. opendatasci/_tui/commands.py +95 -0
  11. opendatasci/_tui/completion.py +139 -0
  12. opendatasci/_tui/controller.py +644 -0
  13. opendatasci/_tui/file_refs.py +153 -0
  14. opendatasci/_tui/models.py +4 -0
  15. opendatasci/_tui/presenter.py +259 -0
  16. opendatasci/_tui/service.py +78 -0
  17. opendatasci/_tui/session.py +53 -0
  18. opendatasci/_tui/styles.tcss +248 -0
  19. opendatasci/_tui/styles_visible.tcss +245 -0
  20. opendatasci/_tui/theme.py +113 -0
  21. opendatasci/_tui/tools_display.py +86 -0
  22. opendatasci/_tui/widgets.py +1001 -0
  23. opendatasci/_utils/__init__.py +0 -0
  24. opendatasci/_utils/async_utils.py +11 -0
  25. opendatasci/_utils/data_formats.py +135 -0
  26. opendatasci/_utils/hash_utils.py +52 -0
  27. opendatasci/_utils/langchain_utils.py +155 -0
  28. opendatasci/_utils/streaming_utils.py +23 -0
  29. opendatasci/agents/__init__.py +12 -0
  30. opendatasci/agents/agents.py +515 -0
  31. opendatasci/agents/agents_factory.py +71 -0
  32. opendatasci/agents/chat_memory.py +397 -0
  33. opendatasci/agents/graphs.py +84 -0
  34. opendatasci/agents/nodes.py +74 -0
  35. opendatasci/agents/states.py +36 -0
  36. opendatasci/agents/turn_memory.py +124 -0
  37. opendatasci/configs.py +275 -0
  38. opendatasci/context/__init__.py +7 -0
  39. opendatasci/context/base.py +56 -0
  40. opendatasci/context/local.py +236 -0
  41. opendatasci/models/__init__.py +7 -0
  42. opendatasci/models/anthropic.py +40 -0
  43. opendatasci/models/aws.py +86 -0
  44. opendatasci/models/factory.py +179 -0
  45. opendatasci/models/google.py +79 -0
  46. opendatasci/models/local.py +79 -0
  47. opendatasci/models/microsoft.py +62 -0
  48. opendatasci/models/openai.py +49 -0
  49. opendatasci/models/providers.py +12 -0
  50. opendatasci/prompts/__init__.py +5 -0
  51. opendatasci/prompts/builders.py +85 -0
  52. opendatasci/prompts/caching.py +42 -0
  53. opendatasci/prompts/message_templates.py +7 -0
  54. opendatasci/prompts/prompt_templates.py +227 -0
  55. opendatasci/resources/skills/competitive_data_science.md +241 -0
  56. opendatasci/resources/skills/data_science.md +55 -0
  57. opendatasci/resources/skills/data_science_education.md +42 -0
  58. opendatasci/resources/skills/deep_learning.md +205 -0
  59. opendatasci/resources/skills/machine_learning.md +68 -0
  60. opendatasci/resources/skills/quantitative_analysis.md +45 -0
  61. opendatasci/sandbox/__init__.py +14 -0
  62. opendatasci/sandbox/_runner.py +114 -0
  63. opendatasci/sandbox/base.py +170 -0
  64. opendatasci/sandbox/srt.py +490 -0
  65. opendatasci/skills/__init__.py +9 -0
  66. opendatasci/skills/base.py +28 -0
  67. opendatasci/skills/local.py +131 -0
  68. opendatasci/streaming/__init__.py +37 -0
  69. opendatasci/streaming/events.py +159 -0
  70. opendatasci/streaming/processors.py +387 -0
  71. opendatasci/tools/__init__.py +58 -0
  72. opendatasci/tools/coding.py +261 -0
  73. opendatasci/tools/critic.py +136 -0
  74. opendatasci/tools/dataset_info.py +391 -0
  75. opendatasci/tools/factory.py +172 -0
  76. opendatasci/tools/mcp.py +179 -0
  77. opendatasci/tools/planning.py +88 -0
  78. opendatasci/tools/skills.py +90 -0
  79. opendatasci/tools/user_interaction.py +54 -0
  80. opendatasci/tools/web.py +236 -0
  81. opendatasci/tools/workers.py +237 -0
  82. opendatasci/tools/workspace.py +55 -0
  83. opendatasci/workspace/__init__.py +9 -0
  84. opendatasci/workspace/base.py +20 -0
  85. opendatasci/workspace/local.py +25 -0
@@ -0,0 +1,629 @@
1
+ Metadata-Version: 2.4
2
+ Name: open-data-sci
3
+ Version: 0.1.0
4
+ Summary: AI agent for data science and machine learning
5
+ Project-URL: Homepage, https://github.com/f4roukb/open-data-sci
6
+ Project-URL: Repository, https://github.com/f4roukb/open-data-sci
7
+ Project-URL: Documentation, https://open-data-sci.readthedocs.io
8
+ Project-URL: Bug Tracker, https://github.com/f4roukb/open-data-sci/issues
9
+ Author-email: Farouk Boukil <b4farouk@gmail.com>
10
+ License: Apache License
11
+ Version 2.0, January 2004
12
+ http://www.apache.org/licenses/
13
+
14
+ TERMS AND CONDITIONS FOR USE, REPRODUCTION, AND DISTRIBUTION
15
+
16
+ 1. Definitions.
17
+
18
+ "License" shall mean the terms and conditions for use, reproduction,
19
+ and distribution as defined by Sections 1 through 9 of this document.
20
+
21
+ "Licensor" shall mean the copyright owner or entity authorized by
22
+ the copyright owner that is granting the License.
23
+
24
+ "Legal Entity" shall mean the union of the acting entity and all
25
+ other entities that control, are controlled by, or are under common
26
+ control with that entity. For the purposes of this definition,
27
+ "control" means (i) the power, direct or indirect, to cause the
28
+ direction or management of such entity, whether by contract or
29
+ otherwise, or (ii) ownership of fifty percent (50%) or more of the
30
+ outstanding shares, or (iii) beneficial ownership of such entity.
31
+
32
+ "You" (or "Your") shall mean an individual or Legal Entity
33
+ exercising permissions granted by this License.
34
+
35
+ "Source" form shall mean the preferred form for making modifications,
36
+ including but not limited to software source code, documentation
37
+ source, and configuration files.
38
+
39
+ "Object" form shall mean any form resulting from mechanical
40
+ transformation or translation of a Source form, including but
41
+ not limited to compiled object code, generated documentation,
42
+ and conversions to other media types.
43
+
44
+ "Work" shall mean the work of authorship, whether in Source or
45
+ Object form, made available under the License, as indicated by a
46
+ copyright notice that is included in or attached to the work
47
+ (an example is provided in the Appendix below).
48
+
49
+ "Derivative Works" shall mean any work, whether in Source or Object
50
+ form, that is based on (or derived from) the Work and for which the
51
+ editorial revisions, annotations, elaborations, or other modifications
52
+ represent, as a whole, an original work of authorship. For the purposes
53
+ of this License, Derivative Works shall not include works that remain
54
+ separable from, or merely link (or bind by name) to the interfaces of,
55
+ the Work and Derivative Works thereof.
56
+
57
+ "Contribution" shall mean any work of authorship, including
58
+ the original version of the Work and any modifications or additions
59
+ to that Work or Derivative Works thereof, that is intentionally
60
+ submitted to Licensor for inclusion in the Work by the copyright owner
61
+ or by an individual or Legal Entity authorized to submit on behalf of
62
+ the copyright owner. For the purposes of this definition, "submitted"
63
+ means any form of electronic, verbal, or written communication sent
64
+ to the Licensor or its representatives, including but not limited to
65
+ communication on electronic mailing lists, source code control systems,
66
+ and issue tracking systems that are managed by, or on behalf of, the
67
+ Licensor for the purpose of discussing and improving the Work, but
68
+ excluding communication that is conspicuously marked or otherwise
69
+ designated in writing by the copyright owner as "Not a Contribution."
70
+
71
+ "Contributor" shall mean Licensor and any individual or Legal Entity
72
+ on behalf of whom a Contribution has been received by Licensor and
73
+ subsequently incorporated within the Work.
74
+
75
+ 2. Grant of Copyright License. Subject to the terms and conditions of
76
+ this License, each Contributor hereby grants to You a perpetual,
77
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
78
+ copyright license to reproduce, prepare Derivative Works of,
79
+ publicly display, publicly perform, sublicense, and distribute the
80
+ Work and such Derivative Works in Source or Object form.
81
+
82
+ 3. Grant of Patent License. Subject to the terms and conditions of
83
+ this License, each Contributor hereby grants to You a perpetual,
84
+ worldwide, non-exclusive, no-charge, royalty-free, irrevocable
85
+ (except as stated in this section) patent license to make, have made,
86
+ use, offer to sell, sell, import, and otherwise transfer the Work,
87
+ where such license applies only to those patent claims licensable
88
+ by such Contributor that are necessarily infringed by their
89
+ Contribution(s) alone or by combination of their Contribution(s)
90
+ with the Work to which such Contribution(s) was submitted. If You
91
+ institute patent litigation against any entity (including a
92
+ cross-claim or counterclaim in a lawsuit) alleging that the Work
93
+ or a Contribution incorporated within the Work constitutes direct
94
+ or contributory patent infringement, then any patent licenses
95
+ granted to You under this License for that Work shall terminate
96
+ as of the date such litigation is filed.
97
+
98
+ 4. Redistribution. You may reproduce and distribute copies of the
99
+ Work or Derivative Works thereof in any medium, with or without
100
+ modifications, and in Source or Object form, provided that You
101
+ meet the following conditions:
102
+
103
+ (a) You must give any other recipients of the Work or
104
+ Derivative Works a copy of this License; and
105
+
106
+ (b) You must cause any modified files to carry prominent notices
107
+ stating that You changed the files; and
108
+
109
+ (c) You must retain, in the Source form of any Derivative Works
110
+ that You distribute, all copyright, patent, trademark, and
111
+ attribution notices from the Source form of the Work,
112
+ excluding those notices that do not pertain to any part of
113
+ the Derivative Works; and
114
+
115
+ (d) If the Work includes a "NOTICE" text file as part of its
116
+ distribution, then any Derivative Works that You distribute must
117
+ include a readable copy of the attribution notices contained
118
+ within such NOTICE file, excluding those notices that do not
119
+ pertain to any part of the Derivative Works, in at least one
120
+ of the following places: within a NOTICE text file distributed
121
+ as part of the Derivative Works; within the Source form or
122
+ documentation, if provided along with the Derivative Works; or,
123
+ within a display generated by the Derivative Works, if and
124
+ wherever such third-party notices normally appear. The contents
125
+ of the NOTICE file are for informational purposes only and
126
+ do not modify the License. You may add Your own attribution
127
+ notices within Derivative Works that You distribute, alongside
128
+ or as an addendum to the NOTICE text from the Work, provided
129
+ that such additional attribution notices cannot be construed
130
+ as modifying the License.
131
+
132
+ You may add Your own copyright statement to Your modifications and
133
+ may provide additional or different license terms and conditions
134
+ for use, reproduction, or distribution of Your modifications, or
135
+ for any such Derivative Works as a whole, provided Your use,
136
+ reproduction, and distribution of the Work otherwise complies with
137
+ the conditions stated in this License.
138
+
139
+ 5. Submission of Contributions. Unless You explicitly state otherwise,
140
+ any Contribution intentionally submitted for inclusion in the Work
141
+ by You to the Licensor shall be under the terms and conditions of
142
+ this License, without any additional terms or conditions.
143
+ Notwithstanding the above, nothing herein shall supersede or modify
144
+ the terms of any separate license agreement you may have executed
145
+ with Licensor regarding such Contributions.
146
+
147
+ 6. Trademarks. This License does not grant permission to use the trade
148
+ names, trademarks, service marks, or product names of the Licensor,
149
+ except as required for reasonable and customary use in describing the
150
+ origin of the Work and reproducing the content of the NOTICE file.
151
+
152
+ 7. Disclaimer of Warranty. Unless required by applicable law or
153
+ agreed to in writing, Licensor provides the Work (and each
154
+ Contributor provides its Contributions) on an "AS IS" BASIS,
155
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
156
+ implied, including, without limitation, any warranties or conditions
157
+ of TITLE, NON-INFRINGEMENT, MERCHANTABILITY, or FITNESS FOR A
158
+ PARTICULAR PURPOSE. You are solely responsible for determining the
159
+ appropriateness of using or redistributing the Work and assume any
160
+ risks associated with Your exercise of permissions under this License.
161
+
162
+ 8. Limitation of Liability. In no event and under no legal theory,
163
+ whether in tort (including negligence), contract, or otherwise,
164
+ unless required by applicable law (such as deliberate and grossly
165
+ negligent acts) or agreed to in writing, shall any Contributor be
166
+ liable to You for damages, including any direct, indirect, special,
167
+ incidental, or consequential damages of any character arising as a
168
+ result of this License or out of the use or inability to use the
169
+ Work (including but not limited to damages for loss of goodwill,
170
+ work stoppage, computer failure or malfunction, or any and all
171
+ other commercial damages or losses), even if such Contributor
172
+ has been advised of the possibility of such damages.
173
+
174
+ 9. Accepting Warranty or Additional Liability. While redistributing
175
+ the Work or Derivative Works thereof, You may choose to offer,
176
+ and charge a fee for, acceptance of support, warranty, indemnity,
177
+ or other liability obligations and/or rights consistent with this
178
+ License. However, in accepting such obligations, You may act only
179
+ on Your own behalf and on Your sole responsibility, not on behalf
180
+ of any other Contributor, and only if You agree to indemnify,
181
+ defend, and hold each Contributor harmless for any liability
182
+ incurred by, or claims asserted against, such Contributor by reason
183
+ of your accepting any such warranty or additional liability.
184
+
185
+ END OF TERMS AND CONDITIONS
186
+
187
+ APPENDIX: How to apply the Apache License to your work.
188
+
189
+ To apply the Apache License to your work, attach the following
190
+ boilerplate notice, with the fields enclosed by brackets "[]"
191
+ replaced with your own identifying information. (Don't include
192
+ the brackets!) The text should be enclosed in the appropriate
193
+ comment syntax for the file format. We also recommend that a
194
+ file or class name and description of purpose be included on the
195
+ same "printed page" as the copyright notice for easier
196
+ identification within third-party archives.
197
+
198
+ Copyright [2026] Farouk Boukil
199
+
200
+ Licensed under the Apache License, Version 2.0 (the "License");
201
+ you may not use this file except in compliance with the License.
202
+ You may obtain a copy of the License at
203
+
204
+ http://www.apache.org/licenses/LICENSE-2.0
205
+
206
+ Unless required by applicable law or agreed to in writing, software
207
+ distributed under the License is distributed on an "AS IS" BASIS,
208
+ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
209
+ See the License for the specific language governing permissions and
210
+ limitations under the License.
211
+ License-File: LICENSE
212
+ Keywords: agent,ai,analytics,data-science,machine-learning
213
+ Classifier: License :: OSI Approved :: Apache Software License
214
+ Classifier: Operating System :: MacOS :: MacOS X
215
+ Classifier: Operating System :: POSIX :: Linux
216
+ Classifier: Programming Language :: Python :: 3.12
217
+ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
218
+ Requires-Python: <3.13,>=3.12
219
+ Requires-Dist: catboost<2.0.0,>=1.2.0
220
+ Requires-Dist: category-encoders<3.0.0,>=2.6.0
221
+ Requires-Dist: connectorx<1.0.0,>=0.3.0
222
+ Requires-Dist: duckduckgo-search<7.0.0,>=6.0.0
223
+ Requires-Dist: httpx<1.0.0,>=0.27.0
224
+ Requires-Dist: imbalanced-learn<1.0.0,>=0.12.0
225
+ Requires-Dist: langchain-anthropic<1.0.0,>=0.2.0
226
+ Requires-Dist: langchain-core<1.0.0,>=0.3.0
227
+ Requires-Dist: langchain-openai<1.0.0,>=0.2.0
228
+ Requires-Dist: langgraph<1.0.0,>=0.2.0
229
+ Requires-Dist: lightgbm<5.0.0,>=4.0.0
230
+ Requires-Dist: lxml<7.0.0,>=6.1.0
231
+ Requires-Dist: matplotlib<4.0.0,>=3.7.0
232
+ Requires-Dist: networkx<4.0.0,>=3.0.0
233
+ Requires-Dist: numpy<2.0.0,>=1.26.0
234
+ Requires-Dist: optuna<4.0.0,>=3.6.0
235
+ Requires-Dist: pandas[excel,feather,parquet]<3.0.0,>=2.0.0
236
+ Requires-Dist: plotly<6.0.0,>=5.15.0
237
+ Requires-Dist: polars[calamine]<2.0.0,>=1.0.0
238
+ Requires-Dist: prophet<2.0.0,>=1.1.0
239
+ Requires-Dist: pydantic-settings<3.0.0,>=2.0.0
240
+ Requires-Dist: pydantic<3.0.0,>=2.0.0
241
+ Requires-Dist: pyod<3.0.0,>=2.0.0
242
+ Requires-Dist: python-dotenv<2.0.0,>=1.0.0
243
+ Requires-Dist: pyyaml<7.0,>=6.0
244
+ Requires-Dist: rich<14.0.0,>=13.0.0
245
+ Requires-Dist: sandbox-runtime<1.0.0,>=0.2.0
246
+ Requires-Dist: scikit-learn<2.0.0,>=1.5.0
247
+ Requires-Dist: scipy<2.0.0,>=1.14.0
248
+ Requires-Dist: seaborn<1.0.0,>=0.12.0
249
+ Requires-Dist: shap<1.0.0,>=0.46.0
250
+ Requires-Dist: statsmodels<1.0.0,>=0.14.0
251
+ Requires-Dist: textual<0.48.0,>=0.47.0
252
+ Requires-Dist: umap-learn<1.0.0,>=0.5.0
253
+ Requires-Dist: xgboost<3.0.0,>=2.0.0
254
+ Requires-Dist: xxhash<4.0.0,>=3.5.0
255
+ Provides-Extra: aws
256
+ Requires-Dist: boto3<2.0.0,>=1.34.0; extra == 'aws'
257
+ Requires-Dist: langchain-aws<1.0.0,>=0.2.0; extra == 'aws'
258
+ Provides-Extra: azure
259
+ Requires-Dist: azure-identity<2.0.0,>=1.15.0; extra == 'azure'
260
+ Provides-Extra: dev
261
+ Requires-Dist: kaggle<2.0.0,>=1.6.0; extra == 'dev'
262
+ Requires-Dist: mypy<2.0.0,>=1.13.0; extra == 'dev'
263
+ Requires-Dist: pandas-stubs<3.0.0,>=2.0.0; extra == 'dev'
264
+ Requires-Dist: pytest-asyncio<1.0.0,>=0.23.0; extra == 'dev'
265
+ Requires-Dist: pytest-cov<5.0.0,>=4.0.0; extra == 'dev'
266
+ Requires-Dist: pytest-xdist<4.0.0,>=3.0.0; extra == 'dev'
267
+ Requires-Dist: pytest<8.0.0,>=7.0.0; extra == 'dev'
268
+ Requires-Dist: ruff<1.0.0,>=0.8.0; extra == 'dev'
269
+ Requires-Dist: types-openpyxl<4.0.0,>=3.1.0; extra == 'dev'
270
+ Requires-Dist: vulture<3.0.0,>=2.0.0; extra == 'dev'
271
+ Provides-Extra: docs
272
+ Requires-Dist: mkdocs-material<10.0.0,>=9.5.0; extra == 'docs'
273
+ Requires-Dist: mkdocs<2.0.0,>=1.6.0; extra == 'docs'
274
+ Requires-Dist: mkdocstrings[python]<1.0.0,>=0.26.0; extra == 'docs'
275
+ Provides-Extra: gcp
276
+ Requires-Dist: langchain-google-vertexai<3.0.0,>=2.0.0; extra == 'gcp'
277
+ Provides-Extra: gemini
278
+ Requires-Dist: langchain-google-genai<3.0.0,>=2.0.0; extra == 'gemini'
279
+ Provides-Extra: jax
280
+ Requires-Dist: flax<1.0.0,>=0.8.0; extra == 'jax'
281
+ Requires-Dist: jax[cpu]<1.0.0,>=0.4.0; extra == 'jax'
282
+ Requires-Dist: optax<1.0.0,>=0.2.0; extra == 'jax'
283
+ Provides-Extra: ollama
284
+ Requires-Dist: langchain-ollama<1.0.0,>=0.2.0; extra == 'ollama'
285
+ Description-Content-Type: text/markdown
286
+
287
+ # OpenDataSci
288
+
289
+ A production-grade AI agent for data science and machine learning. See the [project README](../../README.md) for an overview, benchmark results, and feature descriptions.
290
+
291
+ ## Contents
292
+
293
+ - [Installation](#installation)
294
+ - [Quick Start](#quick-start)
295
+ - [TUI Reference](#tui-reference)
296
+ - [Slash Commands](#slash-commands)
297
+ - [File Attachments](#file-attachments)
298
+ - [Key Bindings](#key-bindings)
299
+ - [Themes](#themes)
300
+ - [Python SDK](#python-sdk)
301
+ - [Models](#models)
302
+ - [Configuration](#configuration)
303
+ - [Environment Variables](#environment-variables)
304
+
305
+ ---
306
+
307
+ ## Installation
308
+
309
+ ```bash
310
+ pip install open-data-sci
311
+ ```
312
+
313
+ **Requirements:**
314
+ - Python: 3.12
315
+ - Platform: macOS or Linux (Windows is not supported)
316
+
317
+ ### System dependencies
318
+
319
+ The sandbox that runs model-generated code shells out to native binaries that `pip` cannot install. Install them with your OS package manager before using the agent:
320
+
321
+ ```bash
322
+ # macOS
323
+ brew install ripgrep
324
+
325
+ # Linux (Debian/Ubuntu)
326
+ sudo apt-get install -y bubblewrap socat ripgrep
327
+
328
+ # Linux (Fedora)
329
+ sudo dnf install -y bubblewrap socat ripgrep
330
+
331
+ # Linux (Arch)
332
+ sudo pacman -S --noconfirm bubblewrap socat ripgrep
333
+ ```
334
+
335
+ If you've cloned the repository, `make install-system-dependencies` runs the right command for your platform automatically.
336
+
337
+ ### Provider extras
338
+
339
+ Install optional extras to unlock additional LLM providers:
340
+
341
+ ```bash
342
+ pip install "open-data-sci[aws]" # AWS Bedrock
343
+ pip install "open-data-sci[gemini]" # Google Gemini (AI Studio)
344
+ pip install "open-data-sci[gcp]" # Google Vertex AI
345
+ pip install "open-data-sci[azure]" # Azure OpenAI
346
+ pip install "open-data-sci[ollama]" # Ollama (local models)
347
+ ```
348
+
349
+ ### Capability extras
350
+
351
+ ```bash
352
+ pip install "open-data-sci[jax]" # Deep learning — JAX, Flax, Optax
353
+ ```
354
+
355
+ The `[jax]` extra is required to use the **Deep Learning** skill. Without it, the agent's sandboxed Python environment has no training framework available.
356
+
357
+ Multiple extras can be combined:
358
+
359
+ ```bash
360
+ pip install "open-data-sci[aws,gemini,jax]"
361
+ ```
362
+
363
+ ---
364
+
365
+ ## Quick Start
366
+
367
+ ### Basic setup
368
+
369
+ Set your API key and point OpenDataSci at your data:
370
+
371
+ ```bash
372
+ export ANTHROPIC_API_KEY=sk-ant-...
373
+ opendatasci data.csv
374
+ ```
375
+
376
+ A `.env` file in the working directory is loaded automatically, so you can also place it there:
377
+
378
+ ```bash
379
+ # .env
380
+ ANTHROPIC_API_KEY=sk-ant-...
381
+ ```
382
+
383
+ To use a different provider, pass `--provider`:
384
+
385
+ ```bash
386
+ opendatasci data.csv --provider openai --api-key sk-...
387
+ opendatasci data.csv --provider ollama --model llama3.2:3b # local, no key needed
388
+ ```
389
+
390
+ ### Setup with a config file
391
+
392
+ For a reusable configuration across projects, create a YAML file and pass it with `--config`. TUI flags always take precedence over values in the file.
393
+
394
+ ```yaml
395
+ # datasci.yaml
396
+ provider: anthropic
397
+ model: claude-sonnet-4-6
398
+ secondary_provider: openai
399
+ secondary_model: gpt-4o-mini
400
+ temperature: 0.1
401
+ thinking_budget: 8000
402
+ ```
403
+
404
+ ```bash
405
+ opendatasci data.csv --config datasci.yaml
406
+ ```
407
+
408
+ Annotated config files for every supported provider are available in [`examples/configs/`](examples/configs/).
409
+
410
+ ### Python SDK
411
+
412
+ ```python
413
+ from opendatasci import create_agent
414
+
415
+ async with create_agent("data.csv") as agent:
416
+ async for event in agent.astream("Summarise this dataset and train a model on the target column."):
417
+ print(event)
418
+ ```
419
+
420
+ ### More examples
421
+
422
+ The [`examples/`](examples/README.md) directory covers TUI walkthroughs, batch scripts, Jupyter notebooks, and annotated config files across every supported provider.
423
+
424
+ ---
425
+
426
+ ## TUI Reference
427
+
428
+ ```
429
+ opendatasci PATH [OPTIONS]
430
+ ```
431
+
432
+ ### Arguments
433
+
434
+ | Argument | Description |
435
+ |----------|-------------|
436
+ | `PATH` | Data file or directory to load into the workspace |
437
+
438
+ ### Options
439
+
440
+ | Flag | Default | Description |
441
+ |------|---------|-------------|
442
+ | `--provider` | `anthropic` | LLM provider for the primary model. Choices: `anthropic`, `openai`, `bedrock`, `gemini`, `vertexai`, `azure`, `ollama`, `openai_compatible_server` |
443
+ | `--model` | *(provider default)* | Primary model name — provider-specific identifier. Omit to use the provider's default (see [Models](#models)) |
444
+ | `--secondary-provider` | *(same as `--provider`)* | Provider for the secondary (auxiliary) model — may differ from `--provider` |
445
+ | `--secondary-model` | *(provider default)* | Secondary model name for lightweight tasks (summarisation, etc.) |
446
+ | `--api-key` | *(env var)* | API key for the primary provider. Falls back to the standard env var for the selected provider |
447
+ | `--theme` | `default` | Colour palette. Choices: `default`, `accessible`, `light`, `solarized`, `dracula`. Run `/themes` inside the TUI for descriptions |
448
+ | `--debug` | `false` | Enable debug output — writes a detailed `opendatasci_debug.log` |
449
+ | `--config` | *(none)* | Path to a YAML file containing `OpenDataSciConfig` fields; explicit TUI flags take precedence |
450
+ | `--list-providers` | | Print all supported providers and their default models, then exit |
451
+ | `--version` | | Print the installed version, then exit |
452
+
453
+ ### Examples
454
+
455
+ ```bash
456
+ # Minimal — analyse a single file with the default Anthropic provider
457
+ opendatasci data.xlsx
458
+
459
+ # Switch provider and primary model
460
+ opendatasci data.csv --provider openai --model gpt-4o
461
+
462
+ # Bedrock with a region
463
+ REGION=us-west-2 opendatasci ./project/ --provider bedrock
464
+
465
+ # Colour-blind safe theme
466
+ opendatasci data.parquet --theme accessible
467
+
468
+ # Mix providers — heavy model on one, lightweight secondary on another
469
+ opendatasci data.csv --provider anthropic --secondary-provider openai --secondary-model gpt-5.4-mini
470
+
471
+ # See all available providers
472
+ opendatasci --list-providers
473
+ ```
474
+
475
+ ---
476
+
477
+ ## Slash Commands
478
+
479
+ Type `/` in the input box to trigger autocomplete. All commands are available at any time.
480
+
481
+ | Command | Description |
482
+ |---------|-------------|
483
+ | `/clear` | Clear conversation context (preserves session variables and loaded data) |
484
+ | `/compact` | Summarise and compress conversation history to free up context |
485
+ | `/help` | Show all available commands |
486
+ | `/ls-workspace` | List all files currently in the workspace |
487
+ | `/models` | Show the primary and secondary model in use |
488
+ | `/reset` | Reset the agent session and reload data from disk |
489
+ | `/stop` | Stop the currently running agent turn (future messages resume from where it left off) |
490
+ | `/themes` | List available colour themes with descriptions |
491
+ | `/exit` | Quit OpenDataSci |
492
+
493
+ ---
494
+
495
+ ## File Attachments
496
+
497
+ Attach files or code snippets to any message using the `@` prefix:
498
+
499
+ ```
500
+ @path/to/file.py # attach an entire file
501
+ @path/to/notebook.ipynb:L10-L40 # attach a specific line range
502
+ ```
503
+
504
+ The agent sees the attached content as structured context inline with your message. Paths are resolved relative to your current working directory.
505
+
506
+ ---
507
+
508
+ ## Key Bindings
509
+
510
+ | Key | Action |
511
+ |-----|--------|
512
+ | `Ctrl+C` (×2) | Quit |
513
+ | `Ctrl+D` | Quit |
514
+ | `Ctrl+R` | Reset session |
515
+ | `Ctrl+L` | Clear conversation |
516
+ | `Escape` | Focus input box |
517
+ | `Tab` / `Shift+Tab` | Cycle through autocomplete suggestions |
518
+ | `↑` / `↓` | Navigate input history or autocomplete |
519
+
520
+ ---
521
+
522
+ ## Themes
523
+
524
+ Select a theme at launch with `--theme`. Run `/themes` inside the TUI to see descriptions.
525
+
526
+ | Name | Description |
527
+ |------|-------------|
528
+ | `default` | Dark background with muted accents (built-in default) |
529
+ | `accessible` | Okabe-Ito palette — colour-blind safe |
530
+ | `light` | Light background with dark text |
531
+ | `solarized` | Solarized Dark by Ethan Schoonover |
532
+ | `dracula` | Dracula — vivid pastels on near-black |
533
+
534
+ ---
535
+
536
+ ## Python SDK
537
+
538
+ The async-first Python API gives full programmatic control over the agent.
539
+
540
+ ### Basic usage
541
+
542
+ ```python
543
+ from opendatasci import create_agent
544
+
545
+ async with create_agent("sales.xlsx") as agent:
546
+ async for event in agent.astream("What is the average revenue by region?"):
547
+ print(event)
548
+ ```
549
+
550
+ ### Custom provider and model
551
+
552
+ ```python
553
+ from opendatasci import OpenDataSciConfig, create_agent
554
+
555
+ config = OpenDataSciConfig(
556
+ provider="openai",
557
+ model="gpt-4o",
558
+ openai_api_key="sk-...",
559
+ temperature=0.2,
560
+ )
561
+
562
+ async with create_agent("data.parquet", config=config) as agent:
563
+ async for event in agent.astream("Train a gradient boosting model on the target column."):
564
+ print(event)
565
+ ```
566
+
567
+ ### `OpenDataSciConfig` reference
568
+
569
+ | Field | Description |
570
+ |-------|-------------|
571
+ | `provider` | LLM provider (`"anthropic"`, `"openai"`, `"bedrock"`, `"gemini"`, `"vertexai"`, `"azure"`, `"ollama"`, `"openai_compatible_server"`) |
572
+ | `model` | Primary model identifier — omit to use the provider default |
573
+ | `secondary_provider` | Provider for the lightweight secondary model — defaults to the primary provider |
574
+ | `secondary_model` | Secondary model identifier — omit to use the provider default |
575
+ | `anthropic_api_key` | Anthropic API key (env: `ANTHROPIC_API_KEY`) |
576
+ | `openai_api_key` | OpenAI / OpenAI-compatible server API key (env: `OPENAI_API_KEY`) |
577
+ | `google_api_key` | Google Gemini API key (env: `GOOGLE_API_KEY`) |
578
+ | `azure_api_key` | Azure OpenAI API key (env: `AZURE_OPENAI_API_KEY`) |
579
+ | `aws_region` | AWS region for Bedrock (env: `REGION`) |
580
+ | `google_cloud_project` | GCP project ID for Vertex AI (env: `GOOGLE_CLOUD_PROJECT`) |
581
+ | `google_cloud_location` | Vertex AI region (env: `GOOGLE_CLOUD_LOCATION`) |
582
+ | `azure_endpoint` | Azure OpenAI resource endpoint URL (env: `AZURE_OPENAI_ENDPOINT`) |
583
+ | `llm_server_base_url` | Custom API base URL — required for `ollama` and `openai_compatible_server` (env: `LLM_SERVER_BASE_URL`) |
584
+
585
+ ---
586
+
587
+ ## Models
588
+
589
+ OpenDataSci supports every major LLM provider. Pass `--provider` to the TUI or set it in `OpenDataSciConfig`.
590
+
591
+ | Provider | Flag | Extra required | Default model |
592
+ |----------|------|----------------|---------------|
593
+ | Anthropic | `anthropic` | *(none — default)* | `claude-sonnet-4-6` |
594
+ | OpenAI | `openai` | *(none)* | `gpt-5.5` |
595
+ | OpenAI-compatible server (e.g. vLLM) | `openai_compatible_server` | *(none)* | `meta-llama/Llama-3.2-3B-Instruct` |
596
+ | AWS Bedrock | `bedrock` | `open-data-sci[aws]` | `us.anthropic.claude-sonnet-4-6` |
597
+ | Google Gemini | `gemini` | `open-data-sci[gemini]` | `gemini-2.5-pro` |
598
+ | Google Vertex AI | `vertexai` | `open-data-sci[gcp]` | `gemini-2.5-pro` |
599
+ | Azure OpenAI | `azure` | `open-data-sci[azure]` | `gpt-4o` |
600
+ | Ollama | `ollama` | `open-data-sci[ollama]` | `llama3.2:3b` |
601
+
602
+ Pass `--list-providers` to print this table from the TUI at any time.
603
+
604
+ ---
605
+
606
+ ## Configuration
607
+
608
+ ### Workspace files
609
+
610
+ Place these files inside your workspace's `.opendatasci/` directory:
611
+
612
+ | Path | Purpose |
613
+ |------|---------|
614
+ | `.opendatasci/mcp.json` | MCP server URLs — connects the agent to external tool servers |
615
+ | `.opendatasci/plans/` | Persisted plan files — auto-managed, one file per planning session |
616
+
617
+ ### Environment variables
618
+
619
+ | Variable | Description |
620
+ |----------|-------------|
621
+ | `ANTHROPIC_API_KEY` | API key for the Anthropic provider |
622
+ | `OPENAI_API_KEY` | API key for the OpenAI provider |
623
+ | `REGION` | Cloud provider region |
624
+ | `LLM_SERVER_BASE_URL` | Custom API base URL — used by `ollama` and `openai_compatible_server` providers |
625
+ | `SKILLS_DIRECTORY` | Path to a directory of user-defined skill files (overrides none by default) |
626
+ | `BUILTIN_SKILLS_DIRECTORY` | Path to the built-in skills directory (defaults to the bundled `resources/skills`) |
627
+ | `CODE_EXEC_TIMEOUT` | Max seconds for a single sandboxed code execution (default: `1800`) |
628
+
629
+ A `.env` file in the working directory is loaded automatically at startup.