lionagi 0.5.0__py3-none-any.whl → 0.5.1__py3-none-any.whl

lionagi/core/session/branch_mixins.py

@@ -288,11 +288,39 @@ class BranchOperationMixin(ABC):
             image_detail=image_detail,
             tool_schemas=tool_schemas,
         )
-        kwargs["messages"] = self.msgs.to_chat_msgs(progress)
-        kwargs["messages"].append(ins.chat_msg)
 
+        progress = progress or self.msgs.progress
+        messages = [self.msgs.messages[i] for i in progress]
+
+        if self.msgs.system and "system" not in imodel.allowed_roles:
+            messages = [msg for msg in messages if msg.role != "system"]
+            first_instruction = None
+
+            if len(messages) == 0:
+                first_instruction = ins.model_copy()
+                first_instruction.guidance = self.msgs.system.system_info + (
+                    first_instruction.guidance or ""
+                )
+                messages.append(first_instruction)
+            elif len(messages) >= 1:
+                first_instruction = messages[0]
+                if not isinstance(first_instruction, Instruction):
+                    raise ValueError(
+                        "First message in progress must be an Instruction or System"
+                    )
+                first_instruction = first_instruction.model_copy()
+                first_instruction.guidance = self.msgs.system.system_info + (
+                    first_instruction.guidance or ""
+                )
+                messages[0] = first_instruction
+
+        else:
+            messages.append(ins)
+
+        kwargs["messages"] = [i.chat_msg for i in messages]
         imodel = imodel or self.imodel
         api_response = None
+
         if isinstance(imodel, LiteiModel):
             api_response = await imodel.invoke(**kwargs)
         elif isinstance(imodel, iModel):
@@ -329,12 +357,22 @@ class BranchOperationMixin(ABC):
         skip_validation: bool = False,
         clear_messages: bool = False,
         invoke_action: bool = True,
+        response_format: (
+            type[BaseModel] | BaseModel
+        ) = None,  # alias of request_model
         **kwargs,
     ):
+        if response_format and request_model:
+            raise ValueError(
+                "Cannot specify both response_format and request_model"
+                "as they are aliases for the same parameter."
+            )
+        request_model = request_model or response_format
+
         imodel = imodel or self.imodel
         retry_imodel = retry_imodel or imodel
         if clear_messages:
-            self.clear_messages()
+            self.msgs.clear_messages()
 
         if num_parse_retries > 5:
             logging.warning(

lionagi/integrations/anthropic_/AnthropicService.py

@@ -111,3 +111,7 @@ class AnthropicService(Service):
         return self.check_rate_limiter(
             model_obj, limit_requests=limit_requests, limit_tokens=limit_tokens
         )
+
+    @property
+    def allowed_roles(self):
+        return ["user", "assistant"]

lionagi/integrations/groq_/GroqService.py

@@ -145,3 +145,7 @@ class GroqService(Service):
             model_obj,
             limit_requests=limit_requests,
         )
+
+    @property
+    def allowed_roles(self):
+        return ["user", "assistant", "system"]

lionagi/integrations/litellm_/imodel.py

@@ -67,3 +67,7 @@ class LiteiModel:
                 v = str(v)
             hashable_items.append((k, v))
         return hash(frozenset(hashable_items))
+
+    @property
+    def allowed_roles(self):
+        return ["user", "assistant", "system"]

lionagi/integrations/ollama_/OllamaService.py

@@ -136,3 +136,7 @@ class OllamaService(Service):
     # List Running Models
     def list_running_models(self):
         return OllamaRequest(endpoint="ps", method="GET")
+
+    @property
+    def allowed_roles(self):
+        return ["user", "assistant", "system"]

lionagi/integrations/openai_/OpenAIModel.py

@@ -412,3 +412,7 @@ class OpenAIModel(BaseModel):
                 "Estimating price currently only supports chat/completions endpoint"
             )
         return estimated_price
+
+    @property
+    def allowed_roles(self):
+        return ["user", "assistant", "system"]

lionagi/integrations/perplexity_/PerplexityService.py

@@ -107,3 +107,7 @@ class PerplexityService(Service):
         return self.check_rate_limiter(
             model_obj, limit_requests=limit_requests, limit_tokens=limit_tokens
         )
+
+    @property
+    def allowed_roles(self):
+        return ["user", "assistant", "system"]

lionagi/service/imodel.py

@@ -9,6 +9,7 @@ from lionagi.service.service_match_util import (
 
 
 class iModel:
+
     def __init__(
         self,
         provider: str | Service,
@@ -20,25 +21,34 @@ class iModel:
         interval_requests: int = None,
         **kwargs,
     ):
-
         if api_key is not None:
             api_key = api_key
         elif api_key_schema is not None:
             api_key = api_key_schema
 
-        if api_key is None:
-            if isinstance(provider, str):
+        if task == "chat":
+            match provider:
+                case "openai":
+                    task = "create_chat_completion"
+                case "anthropic":
+                    task = "create_message"
+                case "groq":
+                    task = "create_chat_completion"
+                case "perplexity":
+                    task = "create_chat_completion"
+
+        if isinstance(provider, str):
+            if api_key is None:
                 match provider:
                     case "openai":
                         api_key = "OPENAI_API_KEY"
                     case "anthropic":
                         api_key = "ANTHROPIC_API_KEY"
-                    case "grok":
-                        api_key = "GROK_API_KEY"
+                    case "groq":
+                        api_key = "GROQ_API_KEY"
                     case "perplexity":
-                        api_key = "PERPLEXITY_API_KEY"
+                        api_key = "PERPLEXIY_API_KEY"
 
-        if isinstance(provider, str):
             self.service = match_service(provider, api_key=api_key, **kwargs)
         elif isinstance(provider, Service):
             self.service = provider
@@ -80,6 +90,7 @@ class iModel:
         self.data_model = self.service.match_data_model(self.task)
 
     def parse_to_data_model(self, **kwargs):
+
         if kwargs.get("model") and self.model:
             if kwargs.get("model") != self.model:
                 raise ValueError(
@@ -106,5 +117,9 @@ class iModel:
     def list_tasks(self):
         return self.service.list_tasks()
 
+    @property
+    def allowed_roles(self):
+        return self.service.allowed_roles
+
 
 __all__ = ["iModel"]

lionagi/service/service.py

@@ -3,6 +3,10 @@ class Service:
     def list_tasks(self):
         pass
 
+    @property
+    def allowed_roles(self):
+        return ["user", "assistant", "system"]
+
 
 def register_service(cls):
     original_init = cls.__init__

lionagi/version.py

@@ -1 +1 @@
-__version__ = "0.5.0"
+__version__ = "0.5.1"

{lionagi-0.5.0.dist-info → lionagi-0.5.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lionagi
-Version: 0.5.0
+Version: 0.5.1
 Summary: An AGentic Intelligence Operating System.
 Author-email: HaiyangLi <quantocean.li@gmail.com>
 License:            Apache License
@@ -246,17 +246,35 @@ LION is designed to be:
 - 🔧 **Flexible**: Build any workflow you need
 - 🚀 **Efficient**: Minimal dependencies, maximum performance
 
+
+
+## Installation
+
+LION maintains minimal dependencies for maximum reliability:
+
+```bash
+uv pip install lionagi
+```
+
+Dependencies:
+- litellm
+- jinja2
+- pandas
+- pillow
+- python-dotenv
+
+
 ## Quick Start
 
 ```python
 from lionagi import iModel, Branch
 
 # Initialize model
-gpt4o = iModel(provider="openai", model="gpt-4o")
+gpt4o = iModel(provider="openai", task="chat", model="gpt-4o")
 
 hunter = Branch(
-    system="you are a hilarious dragon hunter who responds in 10 words rhymes",
-    imodel=gpt4o,
+  system="you are a hilarious dragon hunter who responds in 10 words rhymes",
+  imodel=gpt4o,
 )
 
 # Chat asynchronously
@@ -267,20 +285,199 @@ print(await hunter.communicate("I am a dragon"))
 You claim to be a dragon, oh what a braggin'!
 ```
 
-## Installation
+## 📦 Features
 
-LION maintains minimal dependencies for maximum reliability:
+### 1. Model Agnostic Structured Output
 
-```bash
-uv pip install lionagi
+LION provides a unified interface for interacting with any AI model, regardless of the underlying architecture. This allows you to easily switch between models without changing your code.
+
+```python
+from pydantic import BaseModel
+
+class Joke(BaseModel):
+  joke: str
+
+sonnet = iModel(
+  provider="anthropic",
+  model="claude-3-5-sonnet-20241022",
+  max_tokens=100,                    # max_tokens is required for anthropic models
+)
+
+response = await hunter.communicate(
+  instruction="I am a dragon",
+  response_format=Joke,           # structured output in given pydantic model
+  clear_messages=True,            # refresh the conversation
+  imodel=sonnet,                  # use sonnet model, which doesn't support structured output
+)
+
+print(type(response))
+print(response.joke)
 ```
 
-Dependencies:
-- litellm
-- jinja2
-- pandas
-- pillow
-- python-dotenv
+```
+<class '__main__.Joke'>
+Joke(joke='With fiery claws, dragons hide their laughter flaws!')
+```
+
+
+### 2. Complete Observability
+
+```python
+# using perplexity model
+pplx_small = iModel(
+    provider="perplexity",
+    task="chat/completions",
+    model="llama-3.1-sonar-small-128k-online",
+    max_tokens=1000,
+)
+
+b = await hunter.communicate(
+    instruction="What makes a well-behaved dragon?",
+    clear_messages=True,            # refresh the conversation
+    imodel=pplx_small,                  # use sonnet model
+)
+
+print(b)
+```
+
+```
+A well-behaved dragon is one that's calm and bright,
+No stress or fear, just a peaceful night.
+It's active, not lethargic, with a happy face,
+And behaviors like digging, not a frantic pace.
+It's social, friendly, and never a fright,
+Just a gentle soul, shining with delight
+```
+
+```python
+hunter.msgs.last_response.model_response
+```
+
+```
+{'id': '1be10f4c-0936-4050-ab48-91bd86ab11a5',
+ 'model': 'llama-3.1-sonar-small-128k-online',
+ 'object': 'chat.completion',
+ 'created': 1734369700,
+ 'choices': [{'index': 0,
+   'message': {'role': 'assistant',
+    'content': "A well-behaved dragon is one that's calm and bright,\nNo stress or fear, just a peaceful night.\nIt's active, not lethargic, with a happy face,\nAnd behaviors like digging, not a frantic pace.\nIt's social, friendly, and never a fright,\nJust a gentle soul, shining with delight"},
+   'finish_reason': 'stop',
+   'delta': {'role': 'assistant', 'content': ''}}],
+ 'usage': {'prompt_tokens': 40, 'completion_tokens': 69, 'total_tokens': 109},
+ 'citations': [{'url': 'https://dragonsdiet.com/blogs/dragon-care/15-bearded-dragon-behaviors-and-what-they-could-mean'},
+  {'url': 'https://masterbraeokk.tripod.com/dragons/behavior.html'},
+  {'url': 'https://files.eric.ed.gov/fulltext/ED247607.pdf'},
+  {'url': 'https://www.travelchinaguide.com/intro/social_customs/zodiac/dragon/five-elements.htm'},
+  {'url': 'https://www.travelchinaguide.com/intro/social_customs/zodiac/dragon/'}]}
+```
+
+
+### 3. Easy composition of complex workflows
+
+```python
+# chain of thoughts
+from pydantic import Field
+
+class Reason(BaseModel):
+  reason: str
+  confidence_score: float
+
+class Thought(BaseModel):
+  thought: str
+
+class Analysis(BaseModel):
+  thought: list[Thought] = Field(
+    default_factory=list, 
+    description="concise Chain of thoughts from you, 3 step, each in 8 words"
+  )
+  analysis: str = Field(
+    ...,
+    description="Final analysis of the dragon's psyche in 20 words",
+  )
+  reason: list[Reason] = Field(
+    default_factory=list,
+    description="Concise Reasoning behind the analysis, 3 support, each in 8 words"
+  )
+
+context1 = "I am a dragon, I think therefore I am, I suffer from shiny objects syndrome"
+context2 = "I like food and poetry, I use uv sometimes, it's cool but I am not familiar with pip"
+
+async def analyze(context) -> Analysis:
+  psychologist = Branch(
+    system="you are a renowned dragon psychologist",
+    imodel=gpt4o,
+  )
+  return await psychologist.communicate(
+    instruction="analyze the dragon's psyche using chain of thoughts",
+    guidance="think step by step, reason with logic",
+    context=context,
+    response_format=Analysis,
+  )
+
+```
+
+```python
+result1 = await analyze(context1)
+
+print("\nThoughts:")
+for i in result1.thought:
+    print(i.thought)
+
+print("\nAnalysis:")    
+print(result1.analysis)
+
+print("\nReasoning:")
+for i in result1.reason:
+    print(i.reason)
+```
+
+```
+
+Thoughts:
+Dragons are attracted to shiny objects naturally.
+This suggests a strong affinity for hoarding.
+Reflects the dragon's inherent desire for possession.
+
+Analysis:
+The dragon demonstrates a compulsive hoarding behavior linked to attraction for shiny objects.
+
+Reasoning:
+Shiny objects trigger instinctual hoarding behavior.
+Possession indicates a symbol of power and security.
+Hoarding is reinforced by evolutionary survival mechanisms.
+```
+
+```python
+result2 = await analyze(context2)
+
+print("\nThoughts:")
+for i in result2.thought:
+    print(i.thought)
+
+print("\nAnalysis:")    
+print(result2.analysis)
+
+print("\nReasoning:")
+for i in result2.reason:
+    print(i.reason)
+```
+
+```
+Thoughts:
+Dragon enjoys both food and poetry regularly.
+Dragon uses uv light with frequent interest.
+Dragon is unfamiliar and not comfortable with pip.
+
+Analysis:
+The dragon is curious and exploratory, yet selectively cautious about unfamiliar methodologies.
+
+Reasoning:
+Preference for food and poetry suggests curiosity.
+Frequent uv light use indicates exploratory nature.
+Discomfort with pip usage shows selective caution.
+```
+
+
 
 ## 🌟 Example Workflow
 

{lionagi-0.5.0.dist-info → lionagi-0.5.1.dist-info}/RECORD

@@ -1,6 +1,6 @@
 lionagi/__init__.py,sha256=kIVZqqp61CFddL5BbYynIPwRlg3oUbSVTnWh6QOyhaM,495
 lionagi/settings.py,sha256=r3G-ihH5ehIKwsUdqSGKRwxKpXFj9v6kIaElZKQd7R4,2953
-lionagi/version.py,sha256=LBK46heutvn3KmsCrKIYu8RQikbfnjZaj2xFrXaeCzQ,22
+lionagi/version.py,sha256=eZ1bOun1DDVV0YLOBW4wj2FP1ajReLjbIrGmzN7ASBw,22
 lionagi/core/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
 lionagi/core/_class_registry.py,sha256=srSWefqCS9EZrMvyA8zCrZ9KFvzAhTIj8g6mJG5KlIc,1982
 lionagi/core/action/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
@@ -55,7 +55,7 @@ lionagi/core/models/schema_model.py,sha256=1GNCStg10VWyhqjN4ojxs9mt60T5o8I8_ZX7D
 lionagi/core/models/types.py,sha256=I8khSl0c-O2UgYLMss5akvIFscogjfgKTE2sT6Tg50I,270
 lionagi/core/session/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
 lionagi/core/session/branch.py,sha256=MZXmE7Ne8_CGxgS_Jo5Qak3wJdvut4ZXgF2j8zPMstc,4241
-lionagi/core/session/branch_mixins.py,sha256=Li3Y4_EsqsT0c4OwzM1eh1loalAGELUiByPcKWmISAs,18226
+lionagi/core/session/branch_mixins.py,sha256=OvN29nHieren4YQQ47VQ3ZfxKPBwIR7_lrxcefSwJLc,19766
 lionagi/core/session/session.py,sha256=0DW8HoU_Eqki4UeKIaEuXWBb08N_OkxoDWQyH211M9g,5195
 lionagi/core/session/types.py,sha256=MUGTSa2HWK79p7z-CG22RFP07N5AKnPVNXZwZt_wIvU,202
 lionagi/core/typing/__init__.py,sha256=THyLLjX2WgKkycQPNxXWWl1BVh7XCrQMuAh8WE4UhP8,226
@@ -67,7 +67,7 @@ lionagi/core/typing/typing_.py,sha256=VJj5W6y-JGK1ZzSbyDs4qAuq0cA5wp5HtRgZUsZ50E
 lionagi/integrations/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lionagi/integrations/services.py,sha256=qxffUaPKvp2Bb_LI3Uiyokm7l6ZAbRi0xKxZXCYs67c,498
 lionagi/integrations/anthropic_/AnthropicModel.py,sha256=SOqO0xlmOZ9CMl7aEmQylXjDnHJ3c5oyR7POQLrWWGc,9064
-lionagi/integrations/anthropic_/AnthropicService.py,sha256=AHwsGYkblAMqev-COgqZthAVgFrWBWiaqMmFUcxqm6M,3575
+lionagi/integrations/anthropic_/AnthropicService.py,sha256=Z6fGD2nLEKOm3cDK4Gy0_Vl289KAxX1wswCWsEDpp4E,3656
 lionagi/integrations/anthropic_/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
 lionagi/integrations/anthropic_/anthropic_max_output_token_data.yaml,sha256=u_CT9UAXoNpbAWuPHMxH9g4erZW7XUwY4vdSXyE86t4,203
 lionagi/integrations/anthropic_/anthropic_price_data.yaml,sha256=NCoBo5QFiBvrtztZf2FEyYvtKjIOYDuVMAVUINtp3OY,368
@@ -85,7 +85,7 @@ lionagi/integrations/anthropic_/api_endpoints/messages/response/content_models.p
 lionagi/integrations/anthropic_/api_endpoints/messages/response/response_body.py,sha256=96e2GAAiTGkExRullvAsAOGhPPHNByhnv6DK_wz1j40,3213
 lionagi/integrations/anthropic_/api_endpoints/messages/response/usage_models.py,sha256=s6oe5iOU027M7YPS10upnvcPsuFbTV1ZM00lInHPKvA,695
 lionagi/integrations/groq_/GroqModel.py,sha256=8vIHjzcYoIeRvzXQpBDd-ASx6gXckr5qhmUqJJ9UL1g,11721
-lionagi/integrations/groq_/GroqService.py,sha256=i3Vru3w8ktve9ISacMSWpjrnpJW_swPMzIZgbvHmxhc,4553
+lionagi/integrations/groq_/GroqService.py,sha256=v3bn04njIwV7ysOlfLCmapOhHZJy3bNUTr0i12tFqME,4644
 lionagi/integrations/groq_/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
 lionagi/integrations/groq_/groq_max_output_token_data.yaml,sha256=Y0PbyZ7pyyV1zi7ZJSJGVPsZDhSeN2JMOyghzkqqoKc,559
 lionagi/integrations/groq_/groq_price_data.yaml,sha256=HmN_2-QliKvaC8Ghq7QJAD97ubMYzPSy7EGgqzgCz48,1234
@@ -97,9 +97,9 @@ lionagi/integrations/groq_/api_endpoints/groq_request.py,sha256=u-GJuu0ZsY7jMWaX
 lionagi/integrations/groq_/api_endpoints/match_response.py,sha256=95vRKsR1QYPPmBY36dC5azdKn5UlXNRrTQqTUZro_YM,3756
 lionagi/integrations/groq_/api_endpoints/response_utils.py,sha256=P5kRsGHe-Rx9xejfRcU8q680yotcWLTSaSUuqXATcho,3710
 lionagi/integrations/litellm_/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
-lionagi/integrations/litellm_/imodel.py,sha256=Do8fPqdhZi6ImgfATS68imvAMyVqeouCOH5fTPCJ4Z0,2092
+lionagi/integrations/litellm_/imodel.py,sha256=8TalouW9bb1TBlbPeRJjySHrR6u9Chgru43qmYTIYBE,2183
 lionagi/integrations/ollama_/OllamaModel.py,sha256=5kBYIWShsSpQpSgOxdbRk2_4jmss6Y8iISjUcS3KoWw,8341
-lionagi/integrations/ollama_/OllamaService.py,sha256=hUVnmHEFvQlwaozt6A4xQvBhOkia8AC9cEj9LmSB1i0,3989
+lionagi/integrations/ollama_/OllamaService.py,sha256=bJ4kk1FPjn_kecLzxTJgVj05KZPzF5FclHoDA3jdAlg,4080
 lionagi/integrations/ollama_/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
 lionagi/integrations/ollama_/api_endpoints/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
 lionagi/integrations/ollama_/api_endpoints/api_request.py,sha256=N3frStr_ZJmqu5WMmJXioxnXy5m10bZNwO9vjifcGVw,6843
@@ -126,7 +126,7 @@ lionagi/integrations/ollama_/api_endpoints/model/list_model.py,sha256=OrGdJa0evj
 lionagi/integrations/ollama_/api_endpoints/model/pull_model.py,sha256=fiZJcQSRn73SJA9GdlfPBG5RiMISQwBc0y7S_zAlOGA,923
 lionagi/integrations/ollama_/api_endpoints/model/push_model.py,sha256=yDOMVu3ajdNFT1cuzb4R_3qDxlk0qT4aM2oget3aHZ4,961
 lionagi/integrations/ollama_/api_endpoints/model/show_model.py,sha256=CclV6pEmm5iYM25ePnMAiicVJmZzolDim7BsQoEJAw0,864
-lionagi/integrations/openai_/OpenAIModel.py,sha256=J1X2oYdkXUJRTviAeZrsFe6p8Rqcj3Md8wPaLI8z_Ts,15725
+lionagi/integrations/openai_/OpenAIModel.py,sha256=TOQtliWIV_LsfkfmVK8BHVNMnni-jh60EBN1uj73lCI,15816
 lionagi/integrations/openai_/OpenAIService.py,sha256=P0sGnxeWPKe15_KndqCgEdM74AKVJ6NJsgFGORqqlrk,13507
 lionagi/integrations/openai_/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lionagi/integrations/openai_/openai_max_output_token_data.yaml,sha256=3gd8TGnFjy4AdHzvsV9wZjK_r_o26pe3Th75n5eN2zs,263
@@ -219,7 +219,7 @@ lionagi/integrations/pandas_/search_keywords.py,sha256=AJfN8QVu6rUni6As8AOTnzne_
 lionagi/integrations/pandas_/to_df.py,sha256=3vAOCj0Ib2PZNCblg1oA20PjRIrUXR86FHICQLNhLu0,5757
 lionagi/integrations/pandas_/update_cells.py,sha256=7X1bGcPvnEINrLM_zFCWUXIkrFdMGV3TjoEYBq_SThs,1658
 lionagi/integrations/perplexity_/PerplexityModel.py,sha256=FfOR3WwOgZZEw97mai3D9PJS5C1jsZii2tAX5n0IR8s,9171
-lionagi/integrations/perplexity_/PerplexityService.py,sha256=9Hl3Qcy0r8AqNa94dkWgY5CUSXhSFn4PBVsA6CcEOFY,3550
+lionagi/integrations/perplexity_/PerplexityService.py,sha256=q5Ruy60H3cL2a4LOUjB2JI6bkMwKiWVSlrjpATREwqM,3641
 lionagi/integrations/perplexity_/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
 lionagi/integrations/perplexity_/perplexity_max_output_token_data.yaml,sha256=SY6nDrDRhI_HzEBYHaANR_Mc5GRa0SY9Pw_wRuVBlV4,121
 lionagi/integrations/perplexity_/perplexity_price_data.yaml,sha256=eyp_jZktnEbsEv9VJ0TLNzjp99VdOEA0E-el_dAzCTc,284
@@ -350,9 +350,9 @@ lionagi/protocols/registries/_component_registry.py,sha256=MRA8bAgmjqNsFUnzq1I7J
 lionagi/protocols/registries/_pile_registry.py,sha256=iHiQsd0dapzu7L2YA-VPo7qAw6gjWdDETKe_zh1zhHk,543
 lionagi/service/__init__.py,sha256=xuMab03l9QSeBmq0GXjkTE-Oh2R10l0aful5swKrjas,296
 lionagi/service/complete_request_info.py,sha256=V9cWqmQ-fo952EWCooAvezaM_xuxYU4Xy0Jo_0XJnug,301
-lionagi/service/imodel.py,sha256=gObNmi7tTOZRS639YuAtB1oy8XcXIZUusZJ815LAGm0,3758
+lionagi/service/imodel.py,sha256=laiwc4FvO0jHpf7Q9GrL0VIABCIInkn0ld8XUcbabKo,4191
 lionagi/service/rate_limiter.py,sha256=1cCtKfpOo_8h4ZN-oX5HXtnq0iGjF1rxi8htyu65kMU,3746
-lionagi/service/service.py,sha256=uN9mBrZeV-MDOovqNU4kjRjOhGVtxJOJKTgYNO0QDuc,986
+lionagi/service/service.py,sha256=58FPZzLM85fNm4WgSzmZRwbHHuLUW1K0VagKZ7A2ZAs,1077
 lionagi/service/service_match_util.py,sha256=Re2zJgqsBV9LnDcn5N9pUGMQB_O8-jCTR3ZH6-nxFds,4301
 lionagi/service/service_util.py,sha256=z0tU_wAgWq_gDfm_QeNfDbrsmseumoNxVQ1bEpOCBzg,2985
 lionagi/service/token_calculator.py,sha256=6FQ7GvCq4nBo_7QbNyekYJmQrVIzL0CNwiztK3QSR3M,1486
@@ -367,7 +367,7 @@ lionagi/strategies/sequential_chunk.py,sha256=jG_WZXG-Ra3yd30CmX4b3XeCNAUrZGA2-i
 lionagi/strategies/sequential_concurrent_chunk.py,sha256=H7GShaqYlD5XxNJMG2GdOR4Vl8JHDhZb5jxNq8zY0hI,3365
 lionagi/strategies/types.py,sha256=fEvE4d1H4SeCcXcd2dz3q4k8jFIBtxYzjxDN7eJRLtI,769
 lionagi/strategies/utils.py,sha256=DX1dvxia8cNRqEJJbssJ3mgRzo7kgWCTA4y5DYLCCZE,1281
-lionagi-0.5.0.dist-info/METADATA,sha256=OuNuC0O5DostwQRd5b2H11jASoX_tKE7Q-fuzXRGC-U,17275
-lionagi-0.5.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-lionagi-0.5.0.dist-info/licenses/LICENSE,sha256=VXFWsdoN5AAknBCgFqQNgPWYx7OPp-PFEP961zGdOjc,11288
-lionagi-0.5.0.dist-info/RECORD,,
+lionagi-0.5.1.dist-info/METADATA,sha256=FkuZDqqzBAOOQ-y96QXAK_E__O5tbVZAhxIsQggkL7U,22672
+lionagi-0.5.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+lionagi-0.5.1.dist-info/licenses/LICENSE,sha256=VXFWsdoN5AAknBCgFqQNgPWYx7OPp-PFEP961zGdOjc,11288
+lionagi-0.5.1.dist-info/RECORD,,