mojentic 1.0.0__py3-none-any.whl → 1.0.1__py3-none-any.whl

_examples/streaming.py

@@ -41,7 +41,6 @@ def main():
         num_ctx=32768,
         num_predict=-1
     )
-
     for chunk in stream:
         print(chunk, end='', flush=True)
 

mojentic/llm/chat_session.py

@@ -1,4 +1,4 @@
-from typing import List, Optional
+from typing import Iterator, List, Optional
 
 from mojentic.llm import LLMBroker
 from mojentic.llm.gateways.models import LLMMessage, MessageRole
@@ -78,6 +78,29 @@ class ChatSession:
         self.insert_message(LLMMessage(role=MessageRole.Assistant, content=response))
         return response
 
+    def send_stream(self, query) -> Iterator[str]:
+        """
+        Send a query to the LLM and yield response chunks as they arrive. Records the query and
+        the full assembled response in the ongoing chat session after the stream is consumed.
+
+        Parameters
+        ----------
+        query : str
+            The query to send to the LLM.
+
+        Yields
+        ------
+        str
+            Content chunks from the LLM response as they arrive.
+        """
+        self.insert_message(LLMMessage(role=MessageRole.User, content=query))
+        accumulated = []
+        for chunk in self.llm.generate_stream(self.messages, tools=self.tools, temperature=self.temperature):
+            accumulated.append(chunk)
+            yield chunk
+        self._ensure_all_messages_are_sized()
+        self.insert_message(LLMMessage(role=MessageRole.Assistant, content="".join(accumulated)))
+
     def insert_message(self, message: LLMMessage):
         """
         Add a message onto the end of the chat session. If the total token count exceeds the max context, the oldest

mojentic/llm/chat_session_spec.py

@@ -94,6 +94,46 @@ class DescribeChatSession:
             assert chat_session.messages[1].content == "Query message 2"
             assert chat_session.messages[2].content == INTENDED_RESPONSE_MESSAGE
 
+    class DescribeStreamingSend:
+
+        def should_yield_content_chunks(self, chat_session):
+            chat_session.llm.generate_stream.return_value = iter(["Hello", " world"])
+
+            chunks = list(chat_session.send_stream("Query message"))
+
+            assert chunks == ["Hello", " world"]
+
+        def should_grow_message_history_after_stream_consumed(self, chat_session):
+            chat_session.llm.generate_stream.return_value = iter(["Response"])
+
+            list(chat_session.send_stream("Query message"))
+
+            assert len(chat_session.messages) == 3
+
+        def should_record_full_assembled_response_in_history(self, chat_session):
+            chat_session.llm.generate_stream.return_value = iter(["Hello", " world"])
+
+            list(chat_session.send_stream("Query message"))
+
+            assert chat_session.messages[2].content == "Hello world"
+
+        def should_record_user_message_in_history(self, chat_session):
+            chat_session.llm.generate_stream.return_value = iter(["Response"])
+
+            list(chat_session.send_stream("Query message"))
+
+            assert chat_session.messages[1].role == MessageRole.User
+            assert chat_session.messages[1].content == "Query message"
+
+        def should_respect_context_capacity(self, chat_session):
+            chat_session.llm.generate_stream.return_value = iter(["Response 1"])
+            list(chat_session.send_stream("Query 1"))
+
+            chat_session.llm.generate_stream.return_value = iter(["Response 2"])
+            list(chat_session.send_stream("Query 2"))
+
+            assert len(chat_session.messages) == 3
+
     class DescribeMessageRoles:
         """
         Specifications for message role handling

{mojentic-1.0.0.dist-info → mojentic-1.0.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mojentic
-Version: 1.0.0
+Version: 1.0.1
 Summary: Mojentic is an agentic framework that aims to provide a simple and flexible way to assemble teams of agents to solve complex problems.
 Author-email: Stacey Vetzal <stacey@vetzal.com>
 Project-URL: Homepage, https://github.com/svetzal/mojentic
@@ -11,28 +11,28 @@ Classifier: Operating System :: OS Independent
 Requires-Python: >=3.11
 Description-Content-Type: text/markdown
 License-File: LICENSE.md
-Requires-Dist: pydantic>=2.8.0
+Requires-Dist: pydantic>=2.12.5
 Requires-Dist: structlog>=25.5.0
 Requires-Dist: numpy>=2.3.5
-Requires-Dist: ollama>=0.1.6
-Requires-Dist: openai>=2.8.1
-Requires-Dist: anthropic>=0.74.1
+Requires-Dist: ollama>=0.6.1
+Requires-Dist: openai>=2.12.0
+Requires-Dist: anthropic>=0.75.0
 Requires-Dist: tiktoken>=0.12.0
 Requires-Dist: parsedatetime>=2.6
 Requires-Dist: pytz>=2025.2
 Requires-Dist: serpapi>=0.1.5
 Requires-Dist: colorama>=0.4.6
 Provides-Extra: dev
-Requires-Dist: pytest>=9.0.1; extra == "dev"
+Requires-Dist: pytest>=9.0.2; extra == "dev"
 Requires-Dist: pytest-asyncio>=1.3.0; extra == "dev"
 Requires-Dist: pytest-spec>=5.2.0; extra == "dev"
 Requires-Dist: pytest-cov>=7.0.0; extra == "dev"
-Requires-Dist: pytest-mock>=3.10.0; extra == "dev"
+Requires-Dist: pytest-mock>=3.15.0; extra == "dev"
 Requires-Dist: flake8>=7.3.0; extra == "dev"
-Requires-Dist: bandit>=1.7.0; extra == "dev"
-Requires-Dist: pip-audit>=2.0.0; extra == "dev"
-Requires-Dist: mkdocs>=1.5.2; extra == "dev"
-Requires-Dist: mkdocs-material>=9.6.10; extra == "dev"
+Requires-Dist: bandit>=1.9.2; extra == "dev"
+Requires-Dist: pip-audit>=2.10.0; extra == "dev"
+Requires-Dist: mkdocs>=1.6.1; extra == "dev"
+Requires-Dist: mkdocs-material>=9.7.0; extra == "dev"
 Requires-Dist: mkdocs-llmstxt>=0.4.0; extra == "dev"
 Requires-Dist: mkdocstrings[python]>=0.21.0; extra == "dev"
 Requires-Dist: griffe-fieldz>=0.3.0; extra == "dev"

{mojentic-1.0.0.dist-info → mojentic-1.0.1.dist-info}/RECORD

@@ -34,7 +34,7 @@ _examples/simple_llm_repl.py,sha256=l3Ul67AttvLrteC4Eikos2562ujxH9RAfJbr2WNkx2A,
 _examples/simple_structured.py,sha256=V5c1RfhHb8ay8q7Xek_l8A2KEaIugKd0QHOePny6cQY,1312
 _examples/simple_tool.py,sha256=Yr4y7q4-GyvvHEiRPFqVWcAe14IFHcatqVw5ypuaNZs,1279
 _examples/solver_chat_session.py,sha256=uVBsqXRmyj3Vh0JrfZeCw5GzgwrL3-CR_xZOVTnYXGU,2015
-_examples/streaming.py,sha256=XV9v0FkrE8eI5e4Nilq5-OM0Oox0kBpNaz9eLqWnCXI,1451
+_examples/streaming.py,sha256=7iU0ji_-R6T90UgW47b2eFAcKhq5sAqNJ9kLEdH-m6Y,1450
 _examples/tell_user_example.py,sha256=l3nYYyKh45e4wbMtHONnwmYPPlkSFv5Erh1dEukyUL4,1268
 _examples/tracer_demo.py,sha256=KfHr0mpC3G97tIKo470OVDwYWYE7jRLAj0Bp_gZ1D4M,6984
 _examples/tracer_qt_viewer.py,sha256=AOxupJGzkVyNZ_1LATKFz0s9VN-cdiwJ1waJGtVfUtA,14988
@@ -74,8 +74,8 @@ mojentic/agents/simple_recursive_agent_spec.py,sha256=rcIT2BWjT-sc2WevZ0ts9qi9Fe
 mojentic/context/__init__.py,sha256=RKDcfejikUZMDuFYIfJpmLnoXoRCOCfLjOTiicjq1Yo,80
 mojentic/context/shared_working_memory.py,sha256=Zt9MNGErEkDIUAaHvyhEOiTaEobI9l0MV4Z59lQFBr0,396
 mojentic/llm/__init__.py,sha256=pHWdS6XRdPKhEWv1YpXaD5B5mUPojWM9ncYB-bXI2Qo,484
-mojentic/llm/chat_session.py,sha256=0lhCZDxGfq3ma4m-SmJqfQYdyxPQElSFPQOcqAacFlQ,4005
-mojentic/llm/chat_session_spec.py,sha256=8-jj-EHV2WwWuvo3t8I75kSEAYiG1nR-OEwkkLTi_z0,3872
+mojentic/llm/chat_session.py,sha256=SacT4WLjUuoRpG4puNDdTpinlfEIQI8sC3bs2loFOS8,4909
+mojentic/llm/chat_session_spec.py,sha256=Qek3kFmRYFnuS8vSnrhQ1vnanuatrShCpqUV0ffRi-g,5492
 mojentic/llm/llm_broker.py,sha256=d59MvUBNgVAZbL4T6GUp-tMroTwwmcTJfyOzJSvejAw,16924
 mojentic/llm/llm_broker_spec.py,sha256=N0wSAIakWXn-4cxwG3dPR0MycZNTW-lQl9jWHlchC2w,8662
 mojentic/llm/message_composers.py,sha256=8_5fA-J1I3BZ_0YlgZkQhsn_u7H8yMGEVNYHUPYW1X8,12142
@@ -142,8 +142,8 @@ mojentic/tracer/tracer_system.py,sha256=KPSVIfGVOjSx6Vj_SvrisqJXKT6ddwBc_UCMQC6D
 mojentic/tracer/tracer_system_spec.py,sha256=8hpQlmAWyjUvk7ihy339L0buQ-eH5rluaFvyMl-mSH4,8830
 mojentic/utils/__init__.py,sha256=WvNYbtVeliMZn2sMX53CrOQlQLJBXi4mJNoocG7s_kI,116
 mojentic/utils/formatting.py,sha256=YtXh0aYzLB9GKP8ZD6u1By1OBqPOXUtHirtq0GmHNag,948
-mojentic-1.0.0.dist-info/licenses/LICENSE.md,sha256=txSgV8n5zY1W3NiF5HHsCwlaW0e8We1cSC6TuJUqxXA,1060
-mojentic-1.0.0.dist-info/METADATA,sha256=DSiDA6rxRnPJ_v6VNpjn5_ebXgphpDBzAWgWhgc2kig,8709
-mojentic-1.0.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-mojentic-1.0.0.dist-info/top_level.txt,sha256=Q-BvPQ8Eu1jnEqK8Xkr6A9C8Xa1z38oPZRHuA5MCTqg,19
-mojentic-1.0.0.dist-info/RECORD,,
+mojentic-1.0.1.dist-info/licenses/LICENSE.md,sha256=txSgV8n5zY1W3NiF5HHsCwlaW0e8We1cSC6TuJUqxXA,1060
+mojentic-1.0.1.dist-info/METADATA,sha256=-5_SMLRWZTgDjAMo4DNnkdaGyDIjxxtER_4dOFvlpKk,8711
+mojentic-1.0.1.dist-info/WHEEL,sha256=wUyA8OaulRlbfwMtmQsvNngGrxQHAvkKcvRmdizlJi0,92
+mojentic-1.0.1.dist-info/top_level.txt,sha256=Q-BvPQ8Eu1jnEqK8Xkr6A9C8Xa1z38oPZRHuA5MCTqg,19
+mojentic-1.0.1.dist-info/RECORD,,

{mojentic-1.0.0.dist-info → mojentic-1.0.1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: setuptools (80.9.0)
+Generator: setuptools (80.10.2)
 Root-Is-Purelib: true
 Tag: py3-none-any