ollaagent 0.1.2__tar.gz → 0.1.3__tar.gz

{ollaagent-0.1.2 → ollaagent-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: ollaagent
-Version: 0.1.2
+Version: 0.1.3
 Summary: Local LLM agent powered by ollama — memory, plan mode, subagents
 License-File: LICENSE
 Requires-Python: >=3.11

{ollaagent-0.1.2 → ollaagent-0.1.3}/ollaAgent/agent.py

@@ -358,7 +358,7 @@ def _accumulate_tool_calls(msg: dict[str, Any], accumulated: dict[int, dict]) ->
         idx = tc.get("index", len(accumulated))
         if idx not in accumulated:
             accumulated[idx] = {"name": "", "arguments": ""}
-        fn = tc.get("function", {})
+        fn = tc.get("function") or {}
         accumulated[idx]["name"] += fn.get("name", "")
         raw_args = fn.get("arguments", "")
         if isinstance(raw_args, dict):
@@ -403,11 +403,11 @@ def _stream_response(stream: Any) -> tuple[str, str, dict[int, dict], int]:
 
     with Live(console=console, refresh_per_second=10) as live:
         for chunk in stream:
-            msg = chunk.get("message", {})
-            thinking = msg.get("thinking", "")
+            msg = chunk.get("message") or {}
+            thinking = msg.get("thinking") or ""
             if thinking:
                 thinking_content += thinking
-            content = msg.get("content", "")
+            content = msg.get("content") or ""
             if content:
                 assistant_content += content
                 live.update(Markdown(assistant_content))

{ollaagent-0.1.2 → ollaagent-0.1.3}/ollaAgent/memory.py

@@ -49,7 +49,7 @@ class SessionMemory:
         try:
             with self._path.open(encoding="utf-8") as fh:
                 data = json.load(fh)
-            self._entries = [MemoryEntry(**e) for e in data.get("entries", [])]
+            self._entries = [MemoryEntry(**e) for e in (data.get("entries") or [])]
         except (json.JSONDecodeError, KeyError, ValueError):
             self._entries = []
 
@@ -125,6 +125,6 @@ def load_session(path: Path) -> list[dict]:
     try:
         with path.open(encoding="utf-8") as fh:
             data = json.load(fh)
-        return data.get("messages", [])
+        return data.get("messages") or []
     except (json.JSONDecodeError, KeyError):
         return []

{ollaagent-0.1.2 → ollaagent-0.1.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "ollaagent"
-version = "0.1.2"
+version = "0.1.3"
 description = "Local LLM agent powered by ollama — memory, plan mode, subagents"
 readme = "README.md"
 requires-python = ">=3.11"

{ollaagent-0.1.2 → ollaagent-0.1.3}/tests/test_agent.py

@@ -1,10 +1,12 @@
 from unittest.mock import MagicMock, patch
 
-import ollaAgent.agent as agent
 import pytest
+
+import ollaAgent.agent as agent
 from ollaAgent.agent import (_is_model_available, _parse_subagent_input,
-                        build_dispatch, execute_tool, list_available_models,
-                        run_agentic_loop, trim_by_tokens, trim_messages)
+                             build_dispatch, execute_tool,
+                             list_available_models, run_agentic_loop,
+                             trim_by_tokens, trim_messages)
 from ollaAgent.permissions import PermissionConfig, PermissionMode
 
 _TEST_DISPATCH = build_dispatch(PermissionConfig(mode=PermissionMode.AUTO))
@@ -169,14 +171,28 @@ class TestTrimMessages:
 
 
 def _make_chunk(
-    content="", tool_calls=None, thinking="", done=False, prompt_eval_count=0
+    content="",
+    tool_calls=None,
+    thinking="",
+    done=False,
+    prompt_eval_count=0,
+    null_tool_calls=False,
+    null_message=False,
+    null_content=False,
+    null_thinking=False,
 ):
+    """테스트용 stream chunk 생성.
+
+    null_* 플래그: 키는 존재하지만 값이 null인 실제 API 응답 재현용.
+    """
+    if null_message:
+        return {"message": None}
     msg = {}
-    if content:
-        msg["content"] = content
-    if thinking:
-        msg["thinking"] = thinking
-    if tool_calls:
+    msg["content"] = None if null_content else (content or None)
+    msg["thinking"] = None if null_thinking else (thinking or None)
+    if null_tool_calls:
+        msg["tool_calls"] = None  # 실제 API: "tool_calls": null
+    elif tool_calls:
         msg["tool_calls"] = tool_calls
     chunk = {"message": msg}
     if done:
@@ -295,6 +311,42 @@ class TestAgenticLoop:
         assert "최종 답변입니다." in result
         assert "내부적으로 생각 중..." not in result
 
+    def test_null_tool_calls_in_stream_does_not_raise(self):
+        """실제 API가 tool_calls: null 반환 시 TypeError 없이 정상 처리"""
+        mock_client = MagicMock()
+        mock_client.chat.return_value = _make_stream(
+            _make_chunk(null_tool_calls=True),
+            _make_chunk(content="응답입니다."),
+        )
+
+        messages = [{"role": "user", "content": "안녕"}]
+        result = run_agentic_loop(messages, mock_client, _TEST_DISPATCH)
+        assert "응답입니다." in result
+
+    def test_null_message_in_stream_does_not_raise(self):
+        """실제 API가 message: null 반환 시 정상 처리"""
+        mock_client = MagicMock()
+        mock_client.chat.return_value = _make_stream(
+            _make_chunk(null_message=True),
+            _make_chunk(content="응답입니다."),
+        )
+
+        messages = [{"role": "user", "content": "안녕"}]
+        result = run_agentic_loop(messages, mock_client, _TEST_DISPATCH)
+        assert "응답입니다." in result
+
+    def test_null_content_and_thinking_in_stream_does_not_raise(self):
+        """content, thinking이 null인 chunk 정상 처리"""
+        mock_client = MagicMock()
+        mock_client.chat.return_value = _make_stream(
+            _make_chunk(null_content=True, null_thinking=True),
+            _make_chunk(content="정상 응답."),
+        )
+
+        messages = [{"role": "user", "content": "테스트"}]
+        result = run_agentic_loop(messages, mock_client, _TEST_DISPATCH)
+        assert "정상 응답." in result
+
 
 # ──────────────────────────────────────────
 # Unit Tests: write_file

{ollaagent-0.1.2 → ollaagent-0.1.3}/tests/test_memory.py

@@ -1,7 +1,9 @@
 import json
 
 import pytest
-from ollaAgent.memory import MemoryEntry, SessionMemory, load_session, save_session
+
+from ollaAgent.memory import (MemoryEntry, SessionMemory, load_session,
+                              save_session)
 
 # ──────────────────────────────────────────
 # Unit Tests: MemoryEntry
@@ -119,6 +121,13 @@ class TestSessionMemoryPersistence:
         mem = SessionMemory(path=path)
         assert mem.all() == []
 
+    def test_null_entries_key_gives_empty(self, tmp_path):
+        """JSON에 entries: null 포함 시 빈 리스트로 처리"""
+        path = tmp_path / "null_entries.json"
+        path.write_text(json.dumps({"entries": None}))
+        mem = SessionMemory(path=path)
+        assert mem.all() == []
+
 
 class TestSessionMemoryContextString:
 
@@ -174,6 +183,13 @@ class TestSaveLoadSession:
         result = load_session(path)
         assert result == []
 
+    def test_load_null_messages_key_returns_empty(self, tmp_path):
+        """JSON에 messages: null 포함 시 빈 리스트 반환"""
+        path = tmp_path / "null_messages.json"
+        path.write_text(json.dumps({"messages": None}))
+        result = load_session(path)
+        assert result == []
+
     def test_save_creates_parent_dirs(self, tmp_path):
         path = tmp_path / "a" / "b" / "c" / "sess.json"
         save_session([{"role": "user", "content": "x"}], path)

{ollaagent-0.1.2 → ollaagent-0.1.3}/uv.lock

@@ -187,7 +187,7 @@ wheels = [
 
 [[package]]
 name = "ollaagent"
-version = "0.1.1"
+version = "0.1.2"
 source = { editable = "." }
 dependencies = [
     { name = "ollama" },