promptbuilder 0.4.33__tar.gz → 0.4.34__tar.gz

{promptbuilder-0.4.33/promptbuilder.egg-info → promptbuilder-0.4.34}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: promptbuilder
-Version: 0.4.33
+Version: 0.4.34
 Summary: Library for building prompts for LLMs
 Home-page: https://github.com/kapulkin/promptbuilder
 Author: Kapulkin Stanislav

{promptbuilder-0.4.33 → promptbuilder-0.4.34}/promptbuilder/llm_client/logfire_decorators.py

@@ -7,7 +7,7 @@ import logfire
 
 from promptbuilder.llm_client.utils import inherited_decorator
 from promptbuilder.llm_client.config import GLOBAL_CONFIG
-from promptbuilder.llm_client.types import Response, UsageMetadata
+from promptbuilder.llm_client.types import Response, UsageMetadata, content_to_str
 
 
 P = ParamSpec("P")
@@ -24,9 +24,9 @@ def extract_span_data(self, *args, **kwargs) -> dict[str, Any]:
         logfire_messages.append({"role": "system", "content": system_message})
     for content in messages:
         if content.role == "user":
-            logfire_messages.append({"role": "user", "content": content.as_str()})
+            logfire_messages.append({"role": "user", "content": content_to_str(content)})
         elif content.role == "model":
-            logfire_messages.append({"role": "assistant", "content": content.as_str()})
+            logfire_messages.append({"role": "assistant", "content": content_to_str(content)})
     span_data["request_data"] = {"messages": logfire_messages}
     span_data["messages"] = messages
     

{promptbuilder-0.4.33 → promptbuilder-0.4.34}/promptbuilder/llm_client/types.py

@@ -96,11 +96,13 @@ class Content(BaseModel):
     role: Role | None = None
     
     def as_str(self) -> str:
-        if self.parts is None:
-            return ""
-        else:
-            return "\n".join([(part.text or "") for part in self.parts])
+        return content_to_str(self)
 
+def content_to_str(content: Content) -> str:
+    if content.parts is None:
+        return ""
+    else:
+        return "".join([part.text for part in content.parts if part.text])
 
 class FinishReason(Enum):
   """Output only. The reason why the model stopped generating tokens.

{promptbuilder-0.4.33 → promptbuilder-0.4.34/promptbuilder.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: promptbuilder
-Version: 0.4.33
+Version: 0.4.34
 Summary: Library for building prompts for LLMs
 Home-page: https://github.com/kapulkin/promptbuilder
 Author: Kapulkin Stanislav

{promptbuilder-0.4.33 → promptbuilder-0.4.34}/setup.py

@@ -2,7 +2,7 @@ from setuptools import setup, find_packages
 
 setup(
     name="promptbuilder",
-    version="0.4.33",
+    version="0.4.34",
     packages=find_packages(),
     install_requires=[
         "pydantic",