vedana-core 0.1.0.dev6__tar.gz → 0.6.0.dev1__tar.gz

{vedana_core-0.1.0.dev6 → vedana_core-0.6.0.dev1}/CHANGELOG.md

@@ -1,5 +1,6 @@
-# WIP
+# 2026.01.29 - 0.6.0
 
+* Republish to PyPI
 * remove DataModel loading/caching from vedana-core, rely on vedana-etl for updates
 * change DataModel attributes structure - split anchor/edge attributes
 * optimize DataModel calls

{vedana_core-0.1.0.dev6 → vedana_core-0.6.0.dev1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: vedana-core
-Version: 0.1.0.dev6
+Version: 0.6.0.dev1
 Summary: Semantic Graph RAG App
 Author-email: Andrey Tatarinov <a@tatarinov.co>, Timur Sheydaev <tsheyd@epoch8.co>
 Requires-Python: >=3.12

{vedana_core-0.1.0.dev6 → vedana_core-0.6.0.dev1}/pyproject.toml

@@ -81,7 +81,6 @@ custom_steps = """
     MEMGRAPH_USER: ""
     MEMGRAPH_PWD: ""
     EMBEDDINGS_DIM: "1024"
-    OPENAI_API_KEY: ${{ secrets.OPENAI_API_KEY }}
   run: |
     uv run pytest libs/vedana-core/
 """

vedana_core-0.6.0.dev1/tests/test_llm.py

@@ -0,0 +1,68 @@
+from unittest.mock import AsyncMock, MagicMock
+
+import pytest
+from jims_core.llms.llm_provider import LLMProvider
+from pydantic import BaseModel
+
+from vedana_core.llm import LLM, Tool
+
+
+def mock_msg(content: str | None = None, tool_calls: list | None = None):
+    """Create a mock message with to_dict() method."""
+    m = MagicMock(content=content, tool_calls=tool_calls, role="assistant")
+    m.to_dict.return_value = {"role": "assistant", "content": content}
+    return m
+
+
+def mock_tool_call(name: str, args: str, id: str = "call_1"):
+    """Create a mock tool call."""
+    tc = MagicMock(id=id)
+    tc.function.name = name
+    tc.function.arguments = args
+    return tc
+
+
+@pytest.mark.asyncio
+async def test_llm_completion_with_tools() -> None:
+    llm_provider = LLMProvider()
+
+    tc = mock_tool_call("hello_world", '{"name": "Alice"}')
+    llm_provider.chat_completion_with_tools = AsyncMock(
+        side_effect=[(mock_msg(tool_calls=[tc]), [tc]), (mock_msg("Done!"), [])]
+    )
+
+    llm = LLM(llm_provider=llm_provider, prompt_templates={})
+
+    class HelloWorldArgs(BaseModel):
+        name: str
+
+    res_messages, res_content = await llm.create_completion_with_tools(
+        messages=[{"role": "system", "content": "Call hello_world with name='Alice'."}],
+        tools=[
+            Tool(
+                name="hello_world", description="Says hello.", args_cls=HelloWorldArgs, fn=lambda a: f"Hello, {a.name}!"
+            )
+        ],
+    )
+
+    tool_msg = next(m for m in res_messages if m.get("role") == "tool")
+    assert tool_msg.get("content") == "Hello, Alice!"
+
+
+@pytest.mark.asyncio
+async def test_llm_completion_no_tool_calls() -> None:
+    llm_provider = LLMProvider()
+    llm_provider.chat_completion_with_tools = AsyncMock(return_value=(mock_msg("No tools needed."), []))
+
+    llm = LLM(llm_provider=llm_provider, prompt_templates={})
+
+    class DummyArgs(BaseModel):
+        value: str
+
+    res_messages, res_content = await llm.create_completion_with_tools(
+        messages=[{"role": "user", "content": "Hello"}],
+        tools=[Tool(name="dummy", description="Dummy.", args_cls=DummyArgs, fn=lambda a: a.value)],
+    )
+
+    assert res_content == "No tools needed."
+    assert not any(m.get("role") == "tool" for m in res_messages)

vedana_core-0.1.0.dev6/tests/test_llm.py

@@ -1,42 +0,0 @@
-import litellm
-import pytest
-from jims_core.llms.llm_provider import LLMProvider
-from pydantic import BaseModel
-
-from vedana_core.llm import LLM, Tool
-
-litellm._turn_on_debug()
-
-
-@pytest.fixture(scope="module")
-def vcr_config():
-    return {"filter_headers": ["authorization"], "ignore_hosts": ["test"]}
-
-
-@pytest.mark.asyncio
-async def test_llm_completion_with_tools() -> None:
-    llm_provider = LLMProvider()
-    llm = LLM(llm_provider=llm_provider, prompt_templates={})
-
-    class HelloWorldArgs(BaseModel):
-        name: str
-
-    res_messages, res_content = await llm.create_completion_with_tools(
-        messages=[
-            {
-                "role": "system",
-                "content": "Call a tool 'hell_world' with argument 'name' set to 'Alice'.",
-            },
-        ],
-        tools=[
-            Tool(
-                name="hello_world",
-                description="A tool that says hello to the world.",
-                args_cls=HelloWorldArgs,
-                fn=lambda args: f"Hello, {args.name}!",
-            ),
-        ],
-    )
-
-    tool_call_msg = [msg for msg in res_messages if msg.get("role") == "tool"][0]
-    assert tool_call_msg.get("content") == "Hello, Alice!"