flashlite 0.2.0__py3-none-any.whl → 0.2.1__py3-none-any.whl

flashlite/conversation/multi_agent.py

@@ -333,8 +333,7 @@ class MultiAgentChat:
         """
         if agent_name not in self._agents:
             raise ValueError(
-                f"Unknown agent: {agent_name}. "
-                f"Available agents: {list(self._agents.keys())}"
+                f"Unknown agent: {agent_name}. Available agents: {list(self._agents.keys())}"
             )
 
         agent = self._agents[agent_name]
@@ -372,12 +371,8 @@ class MultiAgentChat:
                 metadata={
                     "model": response.model,
                     "tokens": response.usage.total_tokens if response.usage else None,
-                    "input_tokens": (
-                        response.usage.input_tokens if response.usage else None
-                    ),
-                    "output_tokens": (
-                        response.usage.output_tokens if response.usage else None
-                    ),
+                    "input_tokens": (response.usage.input_tokens if response.usage else None),
+                    "output_tokens": (response.usage.output_tokens if response.usage else None),
                     "latency_ms": round(latency_ms, 1),
                 },
                 visible_to=visible_to,
@@ -395,7 +390,7 @@ class MultiAgentChat:
 
         # Validate structured output if requested
         if response_model is not None:
-            return self._validate_structured(
+            return await self._validate_structured(
                 response=response,
                 response_model=response_model,
                 messages=messages,
@@ -432,8 +427,7 @@ class MultiAgentChat:
         if "response_format" not in extra_kwargs:
             resolved_model = (agent.model or self._default_model or "").lower()
             if any(
-                p in resolved_model
-                for p in ["gpt-4", "gpt-3.5", "claude", "gemini", "mistral"]
+                p in resolved_model for p in ["gpt-4", "gpt-3.5", "claude", "gemini", "mistral"]
             ):
                 extra_kwargs["response_format"] = {"type": "json_object"}
 
@@ -537,14 +531,10 @@ class MultiAgentChat:
 
             if msg.agent_name == agent.name:
                 # Agent's own previous messages
-                messages.append(
-                    assistant_message(msg.content, name=_sanitize_name(agent.name))
-                )
+                messages.append(assistant_message(msg.content, name=_sanitize_name(agent.name)))
             else:
                 # Other agents'/sources' messages with name attribution
-                messages.append(
-                    user_message(msg.content, name=_sanitize_name(msg.agent_name))
-                )
+                messages.append(user_message(msg.content, name=_sanitize_name(msg.agent_name)))
 
         return messages
 
@@ -697,9 +687,7 @@ class MultiAgentChat:
             for line in msg.content.split("\n"):
                 lines.append(f"  {line}")
             if include_metadata and msg.metadata:
-                meta_str = ", ".join(
-                    f"{k}={v}" for k, v in msg.metadata.items() if v is not None
-                )
+                meta_str = ", ".join(f"{k}={v}" for k, v in msg.metadata.items() if v is not None)
                 if meta_str:
                     lines.append(f"  ({meta_str})")
             lines.append("")
@@ -739,6 +727,5 @@ class MultiAgentChat:
 
     def __repr__(self) -> str:
         return (
-            f"MultiAgentChat(agents={list(self._agents.keys())}, "
-            f"messages={len(self._transcript)})"
+            f"MultiAgentChat(agents={list(self._agents.keys())}, messages={len(self._transcript)})"
         )

{flashlite-0.2.0.dist-info → flashlite-0.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: flashlite
-Version: 0.2.0
+Version: 0.2.1
 Summary: Batteries-included wrapper for litellm with rate limiting, retries, templating, and more
 Author-email: ndalton12 <niall.dalton12@gmail.com>
 License-File: LICENSE.md

{flashlite-0.2.0.dist-info → flashlite-0.2.1.dist-info}/RECORD

@@ -11,7 +11,7 @@ flashlite/cache/memory.py,sha256=_A4F7NTR9da2KDQW7fcKnUWrC-W_JpaYmb3d6rovX3w,441
 flashlite/conversation/__init__.py,sha256=zSgC4G697mx3T5bKn8WUEkSaSkMQQeHJsfyLdRUM30w,694
 flashlite/conversation/context.py,sha256=NQMLi5_WiN1zDYaPZTO9uJG_dJ3JJiVmAFfGAPM4X6c,10164
 flashlite/conversation/manager.py,sha256=dSQDgtzNt_6T8S1sHSAXKcS3DoBQ2vI9Ig1PZKaTh48,11644
-flashlite/conversation/multi_agent.py,sha256=tt5gNzUXqnbvp84_aWtcoCIlShPWSCzVa_Jt48Xuvy8,25427
+flashlite/conversation/multi_agent.py,sha256=uxS90wuAAKWQugQXcM_hdNlkhzph3XtKydqvcrz6WpM,25175
 flashlite/core/__init__.py,sha256=nWbMMPED_HsD62hkIYv45DDR6zX2_cDWCMPDTNfqSu4,315
 flashlite/core/completion.py,sha256=NTtAJzJ3ba0N0xVs8lCN5htme0SWEMxYroGjI63crw4,3847
 flashlite/core/messages.py,sha256=-EUtEjFjSNY1Lzfrynb9xtYw4FZRKnfFoYQqgsUcQZQ,3848
@@ -36,7 +36,7 @@ flashlite/templating/registry.py,sha256=wp8RaibHKNyu5q4tCdOXJ0B4tey7bv-c0qb9h1a7
 flashlite/tools/__init__.py,sha256=zpQ5KyvZwZaVvaulnpMmL_JjCnMfD08nD_foI95TjVg,1791
 flashlite/tools/definitions.py,sha256=cqyk6GR1qeMkTPFqsadnJc-YkCG15QVafiaf-OjGYNU,11519
 flashlite/tools/execution.py,sha256=iQC7V3R5Tx19suISnnuaDpjpgl8wURwOHmKZbsHL16s,10814
-flashlite-0.2.0.dist-info/METADATA,sha256=a1YS4nT7UJJD98ibIlguAfWdhUc-2SDN9xQ9jBmjjSA,4293
-flashlite-0.2.0.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-flashlite-0.2.0.dist-info/licenses/LICENSE.md,sha256=z2KZcyoH16ayjxlbeBM01uD-bXn1WTcKFab5ZKBhfJE,1068
-flashlite-0.2.0.dist-info/RECORD,,
+flashlite-0.2.1.dist-info/METADATA,sha256=hP_D4Tgs1v6LqFyM5kYn0PIju0JSEOPokkhWEGcX5LE,4293
+flashlite-0.2.1.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+flashlite-0.2.1.dist-info/licenses/LICENSE.md,sha256=z2KZcyoH16ayjxlbeBM01uD-bXn1WTcKFab5ZKBhfJE,1068
+flashlite-0.2.1.dist-info/RECORD,,