Skip to content

Commit 92266f9

Browse files
committed
fix: update all imports and patch paths from arbitrium to arbitrium_core
1 parent 6090f2e commit 92266f9

File tree

11 files changed

+45
-36
lines changed

11 files changed

+45
-36
lines changed

CLAUDE.md

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@ pre-commit run -a # all quality checks
3030
### Python API
3131

3232
```python
33-
from arbitrium import Arbitrium
33+
from arbitrium_core import Arbitrium
3434

3535
async def main():
3636
arb = await Arbitrium.from_settings({

examples/INTERACTIVE_DEMO.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,7 @@ openai_key = getpass.getpass("OpenAI API Key: ")
150150

151151
### Step 3: Initialize Tournament
152152
```python
153-
from arbitrium import Arbitrium
153+
from arbitrium_core import Arbitrium
154154

155155
arbitrium = await Arbitrium.from_settings(config)
156156
# Health checks all models
@@ -367,7 +367,7 @@ cp config.example.yml config.yml
367367
arbitrium
368368

369369
# Or use programmatically
370-
from arbitrium import Arbitrium
370+
from arbitrium_core import Arbitrium
371371

372372
async def main():
373373
arbitrium = await Arbitrium.from_config("config.yml")

examples/benchmark_comparison.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,7 @@
1212
import asyncio
1313
import time
1414

15-
from arbitrium import Arbitrium
15+
from arbitrium_core import Arbitrium
1616

1717

1818
async def main():

examples/interactive_demo.ipynb

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -86,7 +86,7 @@
8686
"\n",
8787
"# Try to import again after installation\n",
8888
"try:\n",
89-
" import arbitrium\n",
89+
" import arbitrium_core\n",
9090
"\n",
9191
" version = arbitrium.__version__\n",
9292
" print(\n",
@@ -391,7 +391,7 @@
391391
"metadata": {},
392392
"outputs": [],
393393
"source": [
394-
"from arbitrium import Arbitrium\n",
394+
"from arbitrium_core import Arbitrium\n",
395395
"\n",
396396
"print(\"🏁 Initializing Arbitrium Framework...\\n\")\n",
397397
"\n",
@@ -766,7 +766,7 @@
766766
"\n",
767767
"**Programmatic Usage:**\n",
768768
"```python\n",
769-
"from arbitrium import Arbitrium\n",
769+
"from arbitrium_core import Arbitrium\n",
770770
"\n",
771771
"arbitrium = await Arbitrium.from_config(\"config.yml\")\n",
772772
"result, metrics = await arbitrium.run_tournament(\"Your question\")\n",

examples/quickstart.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111

1212
import asyncio
1313

14-
from arbitrium import Arbitrium
14+
from arbitrium_core import Arbitrium
1515

1616

1717
async def main():

examples/single_model.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111

1212
import asyncio
1313

14-
from arbitrium import Arbitrium
14+
from arbitrium_core import Arbitrium
1515

1616

1717
async def main():

examples/tournament_basic.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111

1212
import asyncio
1313

14-
from arbitrium import Arbitrium
14+
from arbitrium_core import Arbitrium
1515

1616

1717
async def main():

examples/tournament_with_kb.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313

1414
import asyncio
1515

16-
from arbitrium import Arbitrium
16+
from arbitrium_core import Arbitrium
1717

1818

1919
async def main():

scripts/discover_ollama_models.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -10,7 +10,7 @@
1010

1111
# Use centralized constant if package is available, otherwise fallback
1212
try:
13-
from arbitrium.support.constants import DEFAULT_OLLAMA_URL
13+
from arbitrium_core.support.constants import DEFAULT_OLLAMA_URL
1414
except ImportError:
1515
DEFAULT_OLLAMA_URL = "http://localhost:11434"
1616

tests/integration/test_e2e_cli.py

Lines changed: 27 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -462,7 +462,7 @@ async def test_get_question_interactive_mode(
462462
await app._initialize_arbitrium()
463463

464464
with patch(
465-
"arbitrium.interfaces.cli.main.async_input",
465+
"arbitrium_core.interfaces.cli.main.async_input",
466466
new_callable=AsyncMock,
467467
return_value="Interactive question?",
468468
):
@@ -493,7 +493,7 @@ async def test_get_question_fallback_to_interactive(
493493
await app._initialize_arbitrium()
494494

495495
with patch(
496-
"arbitrium.interfaces.cli.main.async_input",
496+
"arbitrium_core.interfaces.cli.main.async_input",
497497
new_callable=AsyncMock,
498498
return_value="Fallback question?",
499499
):
@@ -557,9 +557,11 @@ async def test_run_with_keyboard_interrupt(
557557
side_effect=KeyboardInterrupt()
558558
)
559559

560-
with patch("arbitrium.interfaces.cli.main.App._initialize_arbitrium"):
560+
with patch(
561+
"arbitrium_core.interfaces.cli.main.App._initialize_arbitrium"
562+
):
561563
with patch(
562-
"arbitrium.interfaces.cli.main.App._get_app_question",
564+
"arbitrium_core.interfaces.cli.main.App._get_app_question",
563565
new_callable=AsyncMock,
564566
return_value="Test",
565567
):
@@ -591,9 +593,11 @@ async def test_run_with_tournament_error(
591593
side_effect=Exception("Tournament error")
592594
)
593595

594-
with patch("arbitrium.interfaces.cli.main.App._initialize_arbitrium"):
596+
with patch(
597+
"arbitrium_core.interfaces.cli.main.App._initialize_arbitrium"
598+
):
595599
with patch(
596-
"arbitrium.interfaces.cli.main.App._get_app_question",
600+
"arbitrium_core.interfaces.cli.main.App._get_app_question",
597601
new_callable=AsyncMock,
598602
return_value="Test",
599603
):
@@ -629,11 +633,14 @@ def test_run_from_cli_success(
629633
}
630634

631635
with patch(
632-
"arbitrium.interfaces.cli.main.parse_arguments", return_value=args
636+
"arbitrium_core.interfaces.cli.main.parse_arguments",
637+
return_value=args,
633638
):
634-
with patch("arbitrium.support.logging.setup.setup_logging"):
635-
with patch("arbitrium.interfaces.cli.main.colorama.init"):
636-
with patch("arbitrium.interfaces.cli.main.asyncio.run"):
639+
with patch("arbitrium_core.support.logging.setup.setup_logging"):
640+
with patch("arbitrium_core.interfaces.cli.main.colorama.init"):
641+
with patch(
642+
"arbitrium_core.interfaces.cli.main.asyncio.run"
643+
):
637644
run_from_cli()
638645

639646
def test_run_from_cli_fatal_error_handling(self) -> None:
@@ -648,12 +655,13 @@ def test_run_from_cli_fatal_error_handling(self) -> None:
648655
}
649656

650657
with patch(
651-
"arbitrium.interfaces.cli.main.parse_arguments", return_value=args
658+
"arbitrium_core.interfaces.cli.main.parse_arguments",
659+
return_value=args,
652660
):
653-
with patch("arbitrium.support.logging.setup.setup_logging"):
654-
with patch("arbitrium.interfaces.cli.main.colorama.init"):
661+
with patch("arbitrium_core.support.logging.setup.setup_logging"):
662+
with patch("arbitrium_core.interfaces.cli.main.colorama.init"):
655663
with patch(
656-
"arbitrium.interfaces.cli.main.asyncio.run",
664+
"arbitrium_core.interfaces.cli.main.asyncio.run",
657665
side_effect=FatalError("Fatal error"),
658666
):
659667
with patch("sys.exit") as mock_exit:
@@ -672,12 +680,13 @@ def test_run_from_cli_keyboard_interrupt_handling(self) -> None:
672680
}
673681

674682
with patch(
675-
"arbitrium.interfaces.cli.main.parse_arguments", return_value=args
683+
"arbitrium_core.interfaces.cli.main.parse_arguments",
684+
return_value=args,
676685
):
677-
with patch("arbitrium.support.logging.setup.setup_logging"):
678-
with patch("arbitrium.interfaces.cli.main.colorama.init"):
686+
with patch("arbitrium_core.support.logging.setup.setup_logging"):
687+
with patch("arbitrium_core.interfaces.cli.main.colorama.init"):
679688
with patch(
680-
"arbitrium.interfaces.cli.main.asyncio.run",
689+
"arbitrium_core.interfaces.cli.main.asyncio.run",
681690
side_effect=KeyboardInterrupt(),
682691
):
683692
with patch("sys.exit") as mock_exit:

0 commit comments

Comments
 (0)