python 3.10 ++

Author: HendrikStrobelt

docs/examples/agents/react.py

@@ -2,7 +2,7 @@
 import inspect
 import json
 from collections.abc import Callable
-from typing import Literal
+from typing import Literal, Unpack
 
 import pydantic
 from jinja2 import Template
@@ -83,7 +83,7 @@ def call_tool(self, tool: ReactTool, kwargs_json: str):
     def tool_name_schema(self):
         names = self.tool_names()
         fields = dict()
-        fields["tool"] = Literal[*names]
+        fields["tool"] = Literal[Unpack[names]]
         return pydantic.create_model("ToolSelectionSchema", **fields)
 
     def get_tool_from_schema(self, content: str):

docs/examples/agents/react_instruct.py

@@ -2,7 +2,7 @@
 import inspect
 import json
 from collections.abc import Callable
-from typing import Literal
+from typing import Literal, Unpack
 
 import pydantic
 from jinja2 import Template
@@ -81,7 +81,7 @@ def call_tool(self, tool: ReactTool, kwargs_json: str):
     def tool_name_schema(self):
         names = self.tool_names()
         fields = dict()
-        fields["tool"] = Literal[*names]
+        fields["tool"] = Literal[Unpack[names]]
         return pydantic.create_model("ToolSelectionSchema", **fields)
 
     def get_tool_from_schema(self, content: str):

docs/examples/mini_researcher/researcher.py

@@ -232,11 +232,13 @@ def research_subtopic(main_topic: str, subtopic: str, context: list[RAGDocument]
     summaries = step_summarize_docs(
         m, docs=[c.content for c in context], user_args=user_args
     )
-    print(f"Summaries: \n{'\n\n'.join([w(s) for s in summaries])}")
+    summaries_str = "\n\n".join([w(s) for s in summaries])
+    print(f"Summaries: \n{summaries_str}")
 
     # Step 2: Generate Outline
     outline = step_generate_outline(m, user_args=user_args, context=context)
-    print(f"Outline:\n{'\n'.join(outline)}")
+    outline_str = "\n".join(outline)
+    print(f"Outline:\n{outline_str}")
 
     # Step 3: Merge all for the final report
     full_report = step_write_full_report(

docs/examples/rag/simple_rag_with_filter.py

@@ -64,7 +64,8 @@ def is_answer_relevant_to_question(answer: str, question: str) -> bool:
     # Query the index
     print("Query Embedding model...")
     results = query_index(embedding_model, index, query, docs)
-    print(f"results:\n {'\n'.join([f'=> {r}' for r in results])}\n ====")
+    results_str = "\n".join([f"=> {r}" for r in results])
+    print(f"results:\n {results_str}\n ====")
     del embedding_model  # help GC
 
     # Create Mellea session

mellea/backends/openai.py

@@ -18,6 +18,7 @@
 from transformers.tokenization_utils import PreTrainedTokenizer
 
 import mellea.backends.model_ids as model_ids
+from cli.serve.models import ChatCompletionMessage
 from mellea.backends import BaseModelSubclass
 from mellea.backends.aloras import Alora, AloraBackendMixin
 from mellea.backends.formatter import Formatter, FormatterBackend, TemplateFormatter
@@ -557,8 +558,8 @@ def _extract_model_tool_requests(
         calls = chat_response.choices[0].message.tool_calls
         if calls:
             for tool_call in calls:
-                tool_name = tool_call.function.name
-                tool_args = tool_call.function.arguments
+                tool_name = tool_call.function.name  # type: ignore
+                tool_args = tool_call.function.arguments  # type: ignore
 
                 func = tools.get(tool_name)
                 if func is None:

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "pdm.backend"
 
 [project]
 name = "mellea"
-version = "0.0.1"
+version = "0.0.2"
 authors = [
     { name = "Nathan Fulton", email = "[email protected]" },
     { name = "Hendrik Strobelt", email = "[email protected]" },
@@ -17,7 +17,7 @@ authors = [
 ]
 description = "mellea is a library for writing generative programs"
 readme = "README.md"
-requires-python = ">=3.11"
+requires-python = ">=3.10"
 classifiers = [
     "Programming Language :: Python :: 3",
     "License :: OSI Approved :: Apache Software License",
@@ -95,7 +95,7 @@ sphinx = [
 ]
 
 [tool.ruff]
-target-version = "py312"
+target-version = "py310"
 respect-gitignore = true
 
 [tool.ruff.format]