fix StrEnum back compabilty and bump versoin

Author: BeautyyuYanli

pdm.lock

@@ -1,14 +1,14 @@
 [project]
 name = "prompt_bottle"
-version = "0.3.1"
+version = "0.3.2"
 description = "Prompt template engine, built upon Jinja"
 authors = [
     {name = "Yanli",email = "mail@yanli.one"},
 ]
 dependencies = [
     "qwq-tag>=0.1.2",
     "minijinja>=2.11.0",
-    "pydantic-ai-slim>=0.7.2"
+    "pydantic-ai-slim>=1.0.2"
 ]
 requires-python = ">=3.10"
 readme = "README.md"

pyproject.toml

@@ -50,7 +50,7 @@ def stage_split_history(
             isinstance(tag, QwqTag)
             and tag.name == "div"
             and (role := tag.attr.get("role", None))
-            and role in RolesType
+            and any(x.value == role for x in RolesType)
         ):
             yield (
                 RolesType(role),

src/prompt_bottle/pipeline.py

@@ -1,12 +1,19 @@
-from typing import Any
+from typing import TYPE_CHECKING, Any
 
 from pydantic_ai.messages import ModelMessage
 
+if TYPE_CHECKING:
+    from openai.types.chat.chat_completion_message_param import (
+        ChatCompletionMessageParam,
+    )
 
-async def to_openai_chat(source: list[ModelMessage], **model_kwargs: Any):
-    from pydantic_ai.models.openai import OpenAIModel
+
+async def to_openai_chat(
+    source: list[ModelMessage], **model_kwargs: Any
+) -> "list[ChatCompletionMessageParam]":
+    from pydantic_ai.models.openai import OpenAIChatModel
 
     if "model_name" not in model_kwargs:
         model_kwargs["model_name"] = "gpt-4o"
 
-    return await OpenAIModel(**model_kwargs)._map_messages(source) # type: ignore
+    return await OpenAIChatModel(**model_kwargs)._map_messages(source)