aisberg 0.2.0__py3-none-any.whl → 0.2.2__py3-none-any.whl

aisberg/config.py

@@ -1,5 +1,4 @@
 from typing import Union
-
 from pydantic_settings import BaseSettings, SettingsConfigDict
 
 
@@ -15,9 +14,10 @@ class Settings(BaseSettings):
     s3_secret_access_key: Union[str, None] = None
     s3_endpoint: Union[str, None] = None
 
-    # Pour indiquer le fichier .env
-    model_config = SettingsConfigDict(env_file=".env", env_file_encoding="utf-8")
+    # Config Pydantic
+    model_config = SettingsConfigDict(
+        env_file=".env", env_file_encoding="utf-8", extra="ignore"
+    )
 
 
-# Singleton partagé dans tout le SDK
 settings = Settings()

{aisberg-0.2.0.dist-info → aisberg-0.2.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aisberg
-Version: 0.2.0
+Version: 0.2.2
 Summary: Aisberg SDK for Python - A simple and powerful SDK to interact with the Aisberg API
 Author: Free Pro
 Author-email: Mathis Lambert <mathis.lambert@freepro.com>

{aisberg-0.2.0.dist-info → aisberg-0.2.2.dist-info}/RECORD

@@ -1,7 +1,7 @@
 aisberg/__init__.py,sha256=jMX3F2Fh5JobAkRMKGAcOpmNdoxSthgR4v_oM-Yw1l0,141
 aisberg/async_client.py,sha256=0hO2lzdP2Vcb4qV619aJLrsxeXmfYj5wo0mog1bda5Q,3816
 aisberg/client.py,sha256=W1i9mh-o7f283kz2s9B2nHBCegJGN0Lgf1q0lAMNx9I,3853
-aisberg/config.py,sha256=6qSucVFxCBD9TiISDh2bTt7QlrjwvptTwEtcbj4Wt8k,610
+aisberg/config.py,sha256=R1dvL14YeO_0VRr1nUiirWRFN8xY0Nkthurtxnnoc34,587
 aisberg/exceptions.py,sha256=T1ioitOczdkmW3-8XfPG0ta_O2WqzS1iQtzhDBucb5Y,515
 aisberg/utils.py,sha256=4wkrpC5gz8PJWw3kZqaRbd3KekrI37zlDC__JiAVitw,3552
 aisberg/abstract/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -32,15 +32,8 @@ aisberg/modules/workflows.py,sha256=AxyBwCnZkRdgKHUOAoYXMAJH6J7K-qaL1psZ-f-MKhI,
 aisberg/requests/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 aisberg/requests/async_requests.py,sha256=4J6Gj6vY5AD9lXhSTdYdAKQIqsqH9m1uaKc7XLDKWx8,2849
 aisberg/requests/sync_requests.py,sha256=ov7GDgx4C-oQ9xEZ7VQsGVY1fnuqRpPwjS0H_yH1yb8,2839
-aisberg-0.2.0.dist-info/licenses/LICENSE,sha256=dCGgiGa4f14s5mzTYnj1wIZNqGRVG2bUYyyk7nzKlEo,406
-tests/integration/test_collections_integration.py,sha256=HSs-Mhh5JNx2V6EeKo--G3xZq7pMk3z3gLReX7FJidg,3883
-tests/unit/test_collections_sync.py,sha256=DyizG4MIYNtBOC9TlsFdLlH5vz7USA12K9Uydlcydoo,3199
-tmp/test.py,sha256=8H3GxLztspwyt3_wLPtQkwdiLl--qqAn8ZaEUsXbwDE,743
-tmp/test_async.py,sha256=BaY3MkOIDXsersjYgnfaI43kZ42KDj5Andq5u-yO82s,4168
-tmp/test_collection.py,sha256=wV8AnxuaTsHxlzVPigG_qW6kr57lMkKH_bBm9eFfJ-I,1893
-tmp/test_doc_parse.py,sha256=G8-jpZY2cuauzUK9oEvEHjE7Dgjf6LlPFg9F8WyF49g,1188
-tmp/test_sync.py,sha256=87QSz728gvjTXz2JykjF0U_mNsunDYrWsXYDjkBybI8,4929
-aisberg-0.2.0.dist-info/METADATA,sha256=BZzpCVk1OsZQEw9oomWeK8Q50BPVi_YeCvY4JXvPX4E,5604
-aisberg-0.2.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aisberg-0.2.0.dist-info/top_level.txt,sha256=7_CasmLkyF_h1CVRncNuAjam-DMIeMu1Rev_LpVcWoA,18
-aisberg-0.2.0.dist-info/RECORD,,
+aisberg-0.2.2.dist-info/licenses/LICENSE,sha256=dCGgiGa4f14s5mzTYnj1wIZNqGRVG2bUYyyk7nzKlEo,406
+aisberg-0.2.2.dist-info/METADATA,sha256=GsIDsQLP8zBATgBXSXGziMDs2ax9RTMRGEUacFxUJpg,5604
+aisberg-0.2.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aisberg-0.2.2.dist-info/top_level.txt,sha256=Q5EmjFuum10zWgRsk-c4cGKe25F8J7HYvga4AtTwYXQ,8
+aisberg-0.2.2.dist-info/RECORD,,

aisberg-0.2.2.dist-info/top_level.txt

@@ -0,0 +1 @@
+aisberg

aisberg-0.2.0.dist-info/top_level.txt

@@ -1,3 +0,0 @@
-aisberg
-tests
-tmp

tests/integration/test_collections_integration.py

@@ -1,115 +0,0 @@
-import pytest
-import pytest_asyncio
-
-from aisberg.modules import AsyncCollectionsModule, SyncCollectionsModule
-
-
-## Fixtures for asynchronous tests
-@pytest_asyncio.fixture
-async def async_parent():
-    from aisberg import AisbergAsyncClient
-
-    return await AisbergAsyncClient().initialize()
-
-
-@pytest_asyncio.fixture
-async def async_collections_module(async_parent):
-    return AsyncCollectionsModule(parent=async_parent, client=async_parent._client)
-
-
-## Fixtures for synchronous tests
-@pytest.fixture()
-def sync_parent():
-    from aisberg import AisbergClient
-
-    return AisbergClient()
-
-
-@pytest.fixture
-def sync_collections_module(sync_parent):
-    return SyncCollectionsModule(parent=sync_parent, client=sync_parent._client)
-
-
-## ---------------------------------------------------------
-## Tests for asynchronous collections module
-## ---------------------------------------------------------
-@pytest.mark.asyncio
-async def test_async_list_returns_groups(async_collections_module):
-    result = await async_collections_module.list()
-    assert isinstance(result, list)
-    assert len(result) > 0, "Aucune collection trouvée sur l'API réelle."
-    assert hasattr(result[0], "group")
-    assert hasattr(result[0], "collections")
-
-
-@pytest.mark.asyncio
-async def test_async_get_by_group(async_collections_module):
-    groups = await async_collections_module.list()
-    group_id = groups[0].group
-    collections = await async_collections_module.get_by_group(group_id)
-    assert isinstance(collections, list)
-    assert len(collections) > 0
-    assert hasattr(collections[0], "name")
-
-
-@pytest.mark.asyncio
-async def test_async_details(async_collections_module):
-    groups = await async_collections_module.list()
-    group_id = groups[0].group
-    collection = groups[0].collections[0]
-    details = await async_collections_module.details(
-        collection_id=collection.name, group_id=group_id
-    )
-    assert hasattr(details, "name")
-    assert hasattr(details, "group")
-    assert details.name == collection.name
-    assert details.group == group_id
-    assert isinstance(details.points, list)
-
-
-@pytest.mark.asyncio
-async def test_async_get_by_group_not_found(async_collections_module):
-    with pytest.raises(ValueError):
-        await async_collections_module.get_by_group("THIS_GROUP_DOES_NOT_EXIST")
-
-
-## ---------------------------------------------------------
-## Tests for synchronous collections module
-## ---------------------------------------------------------
-def test_sync_list_returns_groups(sync_collections_module):
-    result = sync_collections_module.list()
-    assert isinstance(result, list)
-    assert len(result) > 0, "Aucune collection trouvée sur l'API réelle."
-    assert hasattr(result[0], "group")
-    assert hasattr(result[0], "collections")
-
-
-def test_sync_get_by_group(sync_collections_module):
-    # On prend le premier groupe connu
-    groups = sync_collections_module.list()
-    group_id = groups[0].group
-    collections = sync_collections_module.get_by_group(group_id)
-    assert isinstance(collections, list)
-    assert len(collections) > 0
-    assert hasattr(collections[0], "name")
-
-
-def test_sync_details(sync_collections_module):
-    # On prend le premier groupe et sa première collection
-    groups = sync_collections_module.list()
-    group_id = groups[0].group
-    collection = groups[0].collections[0]
-    details = sync_collections_module.details(
-        collection_id=collection.name, group_id=group_id
-    )
-    assert hasattr(details, "name")
-    assert hasattr(details, "group")
-    assert details.name == collection.name
-    assert details.group == group_id
-    assert isinstance(details.points, list)
-
-
-def test_sync_get_by_group_not_found(sync_collections_module):
-    # On teste un groupe inexistant
-    with pytest.raises(ValueError):
-        sync_collections_module.get_by_group("THIS_GROUP_DOES_NOT_EXIST")

tests/unit/test_collections_sync.py

@@ -1,104 +0,0 @@
-import pytest
-from unittest.mock import patch, MagicMock
-
-from aisberg.models.collections import (
-    GroupCollections,
-    Collection,
-    CollectionDetails,
-)
-from aisberg.modules import SyncCollectionsModule
-
-
-# Données factices pour les tests
-@pytest.fixture
-def mock_collections_data():
-    return [
-        GroupCollections(
-            group="group1",
-            collections=[
-                Collection(name="col1"),
-                Collection(name="col2"),
-            ],
-        ),
-        GroupCollections(
-            group="group2",
-            collections=[
-                Collection(name="col3"),
-            ],
-        ),
-    ]
-
-
-@pytest.fixture
-def mock_points_data():
-    return [
-        {
-            "id": "pt1",
-            "payload": {
-                "method": "m1",
-                "norm": "normA",
-                "filetype": "txt",
-                "filename": "test.txt",
-                "dense_encoder": "encoderA",
-                "Category": "cat1",
-                "text": "Ceci est un texte",
-                "timestamp": "2024-06-21T10:00:00",
-                "collection_name": "col1",
-                "sparse_encoder": "spA",
-            },
-        }
-    ]
-
-
-@pytest.fixture
-def module():
-    # Un parent factice, client factice
-    parent = MagicMock()
-    client = MagicMock()
-    return SyncCollectionsModule(parent, client)
-
-
-def test_list_collections(module, mock_collections_data):
-    with patch("aisberg.api.endpoints.collections", return_value=mock_collections_data):
-        result = module.list()
-        assert isinstance(result, list)
-        assert len(result) == 2
-        assert result[0].group == "group1"
-        assert result[1].collections[0].name == "col3"
-
-
-def test_list_collections_empty(module):
-    with patch("aisberg.api.endpoints.collections", return_value=[]):
-        result = module.list()
-        assert result == []
-
-
-def test_get_by_group_success(module, mock_collections_data):
-    with patch("aisberg.api.endpoints.collections", return_value=mock_collections_data):
-        collections = module.get_by_group("group1")
-        assert isinstance(collections, list)
-        assert len(collections) == 2
-        assert collections[0].name == "col1"
-
-
-def test_get_by_group_not_found(module, mock_collections_data):
-    with patch("aisberg.api.endpoints.collections", return_value=mock_collections_data):
-        with pytest.raises(ValueError, match="No collections found for group ID"):
-            module.get_by_group("groupX")
-
-
-def test_details_success(module, mock_points_data):
-    with patch("aisberg.api.endpoints.collection", return_value=mock_points_data):
-        details = module.details("col1", "group1")
-        assert isinstance(details, CollectionDetails)
-        assert details.name == "col1"
-        assert details.group == "group1"
-        assert len(details.points) == 1
-        assert details.points[0].id == "pt1"
-        assert details.points[0].payload.text == "Ceci est un texte"
-
-
-def test_details_not_found(module):
-    with patch("aisberg.api.endpoints.collection", return_value=None):
-        with pytest.raises(ValueError, match="No collection found"):
-            module.details("col2", "group1")

tmp/test.py

@@ -1,33 +0,0 @@
-from aisberg import AisbergClient, AisbergAsyncClient
-import asyncio
-
-
-client = AisbergClient()
-
-
-# r = client.workflows.run("3de964bb-c98c-4c3a-9b27-8a2c6c8edc57", {})
-# print(r)
-
-
-async def main():
-    async_client = AisbergAsyncClient()
-    # r = await async_client.workflows.run(
-    #     "943925fc-6561-4a01-8450-ada3ad7793f0",
-    #     {
-    #         "input": "Code la suite de Fibonacci en Python",
-    #     },
-    # )
-    # print(r.response)
-    models = await async_client.models.list()
-    print(models)
-
-    res = await async_client.chat.complete(
-        "salut ",
-        model="Qwen/Qwen3-30B-A3B",
-        temperature=0.1,
-    )
-    print(res.choices[0].message.content)
-
-
-if __name__ == "__main__":
-    asyncio.run(main())

tmp/test_async.py

@@ -1,126 +0,0 @@
-from aisberg import AisbergAsyncClient
-from aisberg.models.chat import SystemMessage, HumanMessage, AIMessage
-from aisberg.models.tools import make_tool
-
-aisberg = AisbergAsyncClient()
-
-
-def rag_example():
-    # Example of using the AisbergClient to perform a RAG operation
-    collections = aisberg.collections.get_by_group("free-pro")
-    for collection in collections:
-        print(f"Collection: {collection.name}")
-    chosen_collection = str(input("Enter the collection name to use: "))
-    query = str(input("Enter your query: "))
-
-    print("\n\nRetrieving documents...\n")
-    docs = aisberg.embeddings.retrieve(
-        query=query,
-        collections_names=[chosen_collection],
-        limit=5,
-    )
-
-    print("\nReranking documents...\n")
-    reranked_docs = aisberg.embeddings.rerank(
-        query=query,
-        documents=docs,
-        model="BAAI/bge-reranker-v2-m3",
-        threshold=0.75,  # Keep only 75% accurate documents
-    )
-
-    print("\nGenerating response...\n")
-    response = aisberg.chat.complete(
-        input=f"Answer the following question using the provided documents: {query}\n\nDocuments:\n{reranked_docs}\n\nAnswer:",
-        model="Llama4-Scout-Q4",
-        temperature=0.1,
-    )
-
-    print(response.choices[0].message.content)
-
-
-def chat_example():
-    # Example of using the AisbergClient to perform a chat operation
-    print("\n\nChat with the AI assistant. Type 'exit' to quit.\n")
-
-    history: list = [
-        SystemMessage(
-            content="You are a helpful assistant. You can answer questions and provide information on various topics."
-        )
-    ]
-
-    while True:
-        query = str(input("You : "))
-
-        if query.lower() == "exit":
-            print("Exiting chat. Goodbye!")
-            break
-
-        history.append(HumanMessage(content=query))
-
-        response = aisberg.chat.complete(
-            input=history,
-            model="Llama4-Scout-Q4",
-            temperature=0.1,
-        )
-        print(f"Assistant: {response.choices[0].message.content}")
-        history.append(AIMessage(content=response.choices[0].message.content))
-        print("\n")
-
-
-def tool_call():
-    def get_weather(location: str) -> str:
-        # Simulate a tool call to get weather information
-        return (
-            f"Tool Response : The weather in {location} is sunny with a high of 25°C."
-        )
-
-    def get_shipment_status(shipment_id: str) -> str:
-        return f"Tool Response : The shipment status for {shipment_id} is in transit and expected to arrive on time."
-
-    tools = [
-        make_tool(
-            name="get_weather",
-            description="Get the current weather in a given location",
-            params={"location": "The location to get the weather for"},
-        ),
-        make_tool(
-            name="get_shipment_status",
-            description="Get the shipment status",
-            params={"shipment_id": "The ID of the shipment to check status for"},
-        ),
-    ]
-
-    aisberg.tools.register("get_weather", get_weather)
-    aisberg.tools.register("get_shipment_status", get_shipment_status)
-
-    print("Hello! I can help you with various tasks.")
-    while True:
-        user_input = str(input("> "))
-        response = aisberg.chat.complete(
-            input=[
-                SystemMessage(
-                    content="You are a helpful assistant. To answer questions accurately, you may call external tools when needed. Use `get_weather` to retrieve the current weather for a specific location and `get_shipment_status` to check the status of a shipment by its ID."
-                ),
-                HumanMessage(content=user_input),
-            ],
-            model="Llama4-Scout-Q4",
-            temperature=0.5,
-            tools=tools,
-            auto_execute_tools=True,
-        )
-        print(f"Assistant: {response.choices[0].message.content}")
-
-
-def encode(text: str):
-    embeddings = aisberg.embeddings.encode(
-        input=text,
-        model="BAAI/bge-m3",
-    )
-    print(f"Encoded text: {embeddings}")
-
-
-if __name__ == "__main__":
-    rag_example()
-    # chat_example()
-    # tool_call()
-    # encode("This is a test sentence for encoding.")

tmp/test_collection.py

@@ -1,65 +0,0 @@
-from aisberg import AisbergClient, AisbergAsyncClient
-import json
-
-
-def test_coll():
-    client = AisbergClient()
-    # client.collections.delete('test_aisberg_sdk')
-    #
-    # colls = client.collections.create('test_aisberg_sdk', {
-    #     "chunks": [
-    #         "This is a test chunk 1",
-    #         "This is a test chunk 2",
-    #         "This is a test chunk 3"
-    #     ],
-    #     "metadata": [
-    #         {"source": "test1.txt", "author": "Author 1"},
-    #         {"source": "test2.txt", "author": "Author 2"},
-    #         {"source": "test3.txt", "author": "Author 3"}
-    #     ]
-    # })
-
-
-    points = client.collections.insert_points('test_aisberg_sdk', {
-        "chunks": [
-            "LKFJELZFKJZLEsdfsdfKFJZLKEJF chunk 1",
-            "LKFJELZFKJZsdfsdfLEKFJZLKEJF chunk 2",
-            "LKFJELZFsdfsdfsdfKJZLEKFJZLsdfsdfKEJF chunk 3"
-        ],
-        "metadata": [
-            {"source": "test1.txt", "author": "Author 1"},
-            {"source": "test2.txt", "author": "Author 2"},
-            {"source": "test3.txt", "author": "Author 3"}
-        ]
-    }).points
-
-    print(f"### Collection Created: {points}\n")
-
-    points_to_delete = []
-    for i in range(2):
-        points_to_delete.append(points[i].id)
-    print(f"### Points to delete: {points_to_delete}\n")
-    a = client.collections.delete_points('test_aisberg_sdk', points_to_delete)
-    print(f"### Points Deleted: {a}\n")
-
-    c = client.collections.clear('test_aisberg_sdk')
-    print(f"### Collection Cleared: {c}\n")
-
-
-
-
-async def async_test_coll():
-    client = AisbergAsyncClient()
-    await client.collections.delete('test_aisberg_sdk')
-
-    colls = await client.collections.create('test_aisberg_sdk', './tmp/test.json')
-
-    print(f"### Collection Created: {colls}\n")
-
-
-if __name__ == "__main__":
-    test_coll()
-    #
-    # import asyncio
-    #
-    # asyncio.run(async_test_coll())

tmp/test_doc_parse.py

@@ -1,36 +0,0 @@
-from aisberg import AisbergClient, AisbergAsyncClient
-import json
-
-
-def test_doc():
-    client = AisbergClient()
-    print(client.models.list())
-    stt = str(input("STT model to use (default: tts-1.5): ") or "tts-1.5")
-    docs = client.documents.parse(
-        ["./tmp/test.mp3", "./tmp/test.png"],
-        stt_model=stt,
-        vlm_model="pixtral-12b-2409",
-        input="Que vois tu en haut à droite de l'image ?"
-    )
-
-    print(f"### MP3 Content: {docs[0].content.data}\n")
-    # print(f"### PDF Content: {docs[1].content.data}\n")
-    # print(f"### JSON Content: {json.dumps(docs[2].content.data, indent=2)}\n")
-
-
-async def async_test_doc():
-    client = AisbergAsyncClient()
-    docs = await client.documents.parse(["./tmp/test.mp3", "./tmp/test.pdf", "./tmp/test.json"])
-
-    img = await client.documents.parse("./tmp/test.png", input="Que vois tu en haut à droite de l'image ?")
-
-    print(f"### MP3 Content: {docs[0].content.data}\n")
-    print(f"### PDF Content: {docs[1].content.data}\n")
-    print(f"### JSON Content: {json.dumps(docs[2].content.data, indent=2)}\n")
-
-
-if __name__ == "__main__":
-    test_doc()
-
-    # import asyncio
-    # asyncio.run(async_test_doc())

tmp/test_sync.py

@@ -1,146 +0,0 @@
-from aisberg import AisbergClient
-from aisberg.models.chat import SystemMessage, HumanMessage, AIMessage
-from aisberg.models.tools import make_tool
-
-aisberg = AisbergClient()
-
-
-def rag_example():
-    # Example of using the AisbergClient to perform a RAG operation
-    collections = aisberg.collections.get_by_group("free-pro")
-    for collection in collections:
-        print(f"Collection: {collection.name}")
-    chosen_collection = str(input("Enter the collection name to use: "))
-    query = str(input("Enter your query: "))
-
-    print("\n\nRetrieving documents...\n")
-    docs = aisberg.embeddings.retrieve(
-        query=query,
-        collections_names=[chosen_collection],
-        limit=5,
-    )
-
-    print("\nReranking documents...\n")
-    reranked_docs = aisberg.embeddings.rerank(
-        query=query,
-        documents=docs,
-        model="BAAI/bge-reranker-v2-m3",
-        threshold=0.75,  # Keep only 75% accurate documents
-    )
-
-    print("\nGenerating response...\n")
-    response = aisberg.chat.complete(
-        input=f"Answer the following question using the provided documents: {query}\n\nDocuments:\n{reranked_docs}\n\nAnswer:",
-        model="Llama4-Scout-Q4",
-        temperature=0.1,
-    )
-
-    print(response.choices[0].message.content)
-
-
-def chat_example():
-    # Example of using the AisbergClient to perform a chat operation
-    print("\n\nChat with the AI assistant. Type 'exit' to quit.\n")
-
-    history: list = [
-        SystemMessage(
-            content="You are a helpful assistant. You can answer questions and provide information on various topics."
-        )
-    ]
-
-    while True:
-        query = str(input("You : "))
-
-        if query.lower() == "exit":
-            print("Exiting chat. Goodbye!")
-            break
-
-        history.append(HumanMessage(content=query))
-
-        response = aisberg.chat.complete(
-            input=history,
-            model="Llama4-Scout-Q4",
-            temperature=0.1,
-        )
-        print(f"Assistant: {response.choices[0].message.content}")
-        history.append(AIMessage(content=response.choices[0].message.content))
-        print("\n")
-
-
-def tool_call():
-    def get_weather(location: str) -> str:
-        # Simulate a tool call to get weather information
-        return (
-            f"Tool Response : The weather in {location} is sunny with a high of 25°C."
-        )
-
-    def get_shipment_status(shipment_id: str) -> str:
-        return f"Tool Response : The shipment status for {shipment_id} is in transit and expected to arrive on time."
-
-    tools = [
-        make_tool(
-            name="get_weather",
-            description="Get the current weather in a given location",
-            params={"location": "The location to get the weather for"},
-        ),
-        make_tool(
-            name="get_shipment_status",
-            description="Get the shipment status",
-            params={"shipment_id": "The ID of the shipment to check status for"},
-        ),
-    ]
-
-    aisberg.tools.register("get_weather", get_weather)
-    aisberg.tools.register("get_shipment_status", get_shipment_status)
-
-    print("Hello! I can help you with various tasks.")
-    while True:
-        user_input = str(input("> "))
-        response = aisberg.chat.complete(
-            input=[
-                SystemMessage(
-                    content="You are a helpful assistant. To answer questions accurately, you may call external tools when needed. Use `get_weather` to retrieve the current weather for a specific location and `get_shipment_status` to check the status of a shipment by its ID."
-                ),
-                HumanMessage(content=user_input),
-            ],
-            model="Llama4-Scout-Q4",
-            temperature=0.5,
-            tools=tools,
-            auto_execute_tools=True,
-        )
-        print(f"Assistant: {response.choices[0].message.content}")
-
-
-def encode(text: str):
-    embeddings = aisberg.embeddings.encode(
-        input=text,
-        model="BAAI/bge-m3",
-    )
-    print(f"Encoded text: {embeddings}")
-
-
-if __name__ == "__main__":
-    import sys
-
-    argv = sys.argv[1:]
-    if len(argv) > 0:
-        match argv[0]:
-            case "rag":
-                rag_example()
-            case "chat":
-                chat_example()
-            case "tool":
-                tool_call()
-            case "encode":
-                encode("This is a test sentence for encoding.")
-            case _:
-                print("Usage: python test_sync.py [rag|chat|tool|encode]")
-
-    else:
-        print("Usage: python test_sync.py [rag|chat|tool|encode]")
-        print("Available commands:")
-        print("  rag    - Run a RAG example")
-        print("  chat   - Start a chat session with the AI assistant")
-        print("  tool   - Demonstrate tool calling in chat")
-        print("  encode - Encode a sample text using embeddings")
-        print("Run with 'python test_sync.py [command]' to execute a specific example.")