lionagi 0.8.4__py3-none-any.whl → 0.8.5__py3-none-any.whl

lionagi/operations/ReAct/ReAct.py

@@ -21,6 +21,10 @@ async def ReAct(
     branch: "Branch",
     instruct: Instruct | dict[str, Any],
     interpret: bool = False,
+    interpret_domain: str | None = None,
+    interpret_style: str | None = None,
+    interpret_sample: str | None = None,
+    interpret_kwargs: dict | None = None,
     tools: Any = None,
     tool_schemas: Any = None,
     response_format: type[BaseModel] | BaseModel = None,
@@ -43,7 +47,11 @@ async def ReAct(
                 instruct.to_dict()
                 if isinstance(instruct, Instruct)
                 else instruct
-            )
+            ),
+            domain=interpret_domain,
+            style=interpret_style,
+            sample_writing=interpret_sample,
+            **(interpret_kwargs or {}),
         )
 
     # Convert Instruct to dict if necessary

lionagi/service/endpoints/base.py

@@ -61,6 +61,7 @@ class EndpointConfig(BaseModel):
         use_enum_values=True,
     )
 
+    name: str | None = None
     provider: str | None = None
     base_url: str | None = None
     endpoint: str
@@ -75,6 +76,7 @@ class EndpointConfig(BaseModel):
     requires_tokens: bool = False
     api_version: str | None = None
     allowed_roles: list[str] | None = None
+    request_options: type | None = None
 
 
 class EndPoint(ABC):
@@ -100,6 +102,11 @@ class EndPoint(ABC):
         config.update(kwargs)
         self.config = EndpointConfig(**config)
 
+    @property
+    def name(self) -> str | None:
+        """str | None: The name of the endpoint, if any."""
+        return self.config.name or self.endpoint
+
     @property
     def is_streamable(self) -> bool:
         """bool: Whether this endpoint supports streaming responses."""
@@ -185,6 +192,10 @@ class EndPoint(ABC):
         """bool: Indicates if this endpoint uses role-based messages."""
         return self.allowed_roles is not None
 
+    @property
+    def request_options(self) -> type | None:
+        return self.config.request_options
+
     def create_payload(self, **kwargs) -> dict:
         """Generates a request payload (and headers) for this endpoint.
 

lionagi/service/imodel.py

@@ -5,6 +5,8 @@
 import os
 import warnings
 
+from pydantic import BaseModel
+
 from .endpoints.base import APICalling, EndPoint
 from .endpoints.match_endpoint import match_endpoint
 from .endpoints.rate_limited_processor import RateLimitedAPIExecutor
@@ -266,6 +268,15 @@ class iModel:
         """
         return self.kwargs.get("model", "")
 
+    @property
+    def request_options(self) -> type[BaseModel] | None:
+        """type[BaseModel] | None: The request options model for the endpoint.
+
+        Returns:
+            The request options model if available; otherwise, None.
+        """
+        return self.endpoint.request_options
+
     def to_dict(self):
         kwargs = self.kwargs
         if "kwargs" in self.kwargs:

lionagi/service/providers/exa_/search.py

@@ -1,10 +1,8 @@
-from typing import TYPE_CHECKING, Literal
+from typing import Literal
 
 from lionagi.service.endpoints.base import EndPoint
 
-if TYPE_CHECKING:
-    from .models import ExaSearchRequest
-
+from .models import ExaSearchRequest
 
 CATEGORY_OPTIONS = Literal[
     "article",
@@ -21,6 +19,7 @@ CATEGORY_OPTIONS = Literal[
 ]
 
 SEARCH_CONFIG = {
+    "name": "search_exa",
     "provider": "exa",
     "base_url": "https://api.exa.ai",
     "endpoint": "search",
@@ -47,6 +46,7 @@ SEARCH_CONFIG = {
         "type",  # keyword, neural, auto
         "useAutoPrompt",
     },
+    "request_options": ExaSearchRequest,
 }
 
 

lionagi/service/providers/perplexity_/chat_completions.py

@@ -4,7 +4,10 @@
 
 from lionagi.service.endpoints.chat_completion import ChatCompletionEndPoint
 
+from .models import PerplexityChatCompletionRequest
+
 CHAT_COMPLETION_CONFIG = {
+    "name": "search_perplexity",
     "provider": "perplexity",
     "base_url": "https://api.perplexity.ai",
     "endpoint": "chat/completions",
@@ -31,6 +34,7 @@ CHAT_COMPLETION_CONFIG = {
         "frequency_penalty",
     },
     "allowed_roles": ["user", "assistant"],
+    "request_options": PerplexityChatCompletionRequest,
 }
 
 

lionagi/service/providers/perplexity_/models.py

@@ -0,0 +1,144 @@
+from enum import Enum
+from typing import Any
+
+from pydantic import BaseModel, Field, model_validator
+
+
+class PerplexityRole(str, Enum):
+    """Roles allowed in Perplexity's messages."""
+
+    system = "system"
+    user = "user"
+    assistant = "assistant"
+
+
+class PerplexityMessage(BaseModel):
+    """
+    A single message in the conversation.
+    `role` can be 'system', 'user', or 'assistant'.
+    `content` is the text for that conversation turn.
+    """
+
+    role: PerplexityRole = Field(
+        ...,
+        description="The role of the speaker. Must be system, user, or assistant.",
+    )
+    content: str = Field(..., description="The text content of this message.")
+
+
+class PerplexityChatCompletionRequest(BaseModel):
+    """
+    Represents the request body for Perplexity's Chat Completions endpoint.
+    Endpoint: POST https://api.perplexity.ai/chat/completions
+    """
+
+    model: str = Field(
+        ...,
+        description="The model name, e.g. 'llama-3.1-sonar-small-128k-online'.",
+    )
+    messages: list[PerplexityMessage] = Field(
+        ..., description="A list of messages forming the conversation so far."
+    )
+
+    # Optional parameters
+    frequency_penalty: float | None = Field(
+        default=None,
+        gt=0,
+        description=(
+            "Multiplicative penalty > 0. Values > 1.0 penalize repeated tokens more strongly. "
+            "Value=1.0 means no penalty. Incompatible with presence_penalty."
+        ),
+    )
+    presence_penalty: float | None = Field(
+        default=None,
+        ge=-2.0,
+        le=2.0,
+        description=(
+            "Penalizes tokens that have appeared so far (range -2 to 2). "
+            "Positive values encourage talking about new topics. Incompatible with frequency_penalty."
+        ),
+    )
+    max_tokens: int | None = Field(
+        default=None,
+        description=(
+            "Maximum number of completion tokens. If omitted, model generates tokens until it "
+            "hits stop or context limit."
+        ),
+    )
+    return_images: bool | None = Field(
+        default=None,
+        description="If True, attempt to return images (closed beta feature).",
+    )
+    return_related_questions: bool | None = Field(
+        default=None,
+        description="If True, attempt to return related questions (closed beta feature).",
+    )
+    search_domain_filter: list[Any] | None = Field(
+        default=None,
+        description=(
+            "List of domains to limit or exclude in the online search. Example: ['example.com', '-twitter.com']. "
+            "Supports up to 3 entries. (Closed beta feature.)"
+        ),
+    )
+    search_recency_filter: str | None = Field(
+        default=None,
+        description=(
+            "Returns search results within a specified time interval: 'month', 'week', 'day', or 'hour'."
+        ),
+    )
+    stream: bool | None = Field(
+        default=None,
+        description=(
+            "If True, response is returned incrementally via Server-Sent Events (SSE)."
+        ),
+    )
+    temperature: float | None = Field(
+        default=None,
+        ge=0.0,
+        lt=2.0,
+        description=(
+            "Controls randomness of sampling, range [0, 2). Higher => more random. "
+            "Defaults to 0.2."
+        ),
+    )
+    top_k: int | None = Field(
+        default=None,
+        ge=0,
+        le=2048,
+        description=(
+            "Top-K filtering. 0 disables top-k filtering. If set, only the top K tokens are considered. "
+            "We recommend altering either top_k or top_p, but not both."
+        ),
+    )
+    top_p: float | None = Field(
+        default=None,
+        ge=0.0,
+        le=1.0,
+        description=(
+            "Nucleus sampling threshold. We recommend altering either top_k or top_p, but not both."
+        ),
+    )
+
+    @model_validator(mode="before")
+    def validate_penalties(cls, values):
+        """
+        Disallow using both frequency_penalty != 1.0 and presence_penalty != 0.0 at once,
+        since the docs say they're incompatible.
+        """
+        freq_pen = values.get("frequency_penalty", 1.0)
+        pres_pen = values.get("presence_penalty", 0.0)
+
+        # The doc states frequency_penalty is incompatible with presence_penalty.
+        # We'll enforce that if presence_penalty != 0, frequency_penalty must be 1.0
+        # or vice versa. Adjust logic as needed.
+        if pres_pen != 0.0 and freq_pen != 1.0:
+            raise ValueError(
+                "presence_penalty is incompatible with frequency_penalty. "
+                "Please use only one: either presence_penalty=0 with freq_pen !=1, "
+                "or presence_penalty!=0 with freq_pen=1."
+            )
+        return values
+
+    def to_dict(self) -> dict:
+        """Return a dict suitable for JSON serialization and sending to Perplexity API."""
+        return self.model_dump(exclude_none=True)

lionagi/session/branch.py

@@ -45,6 +45,7 @@ from lionagi.protocols.types import (
     SenderRecipient,
     System,
 )
+from lionagi.service.endpoints.base import EndPoint
 from lionagi.service.types import iModel, iModelManager
 from lionagi.settings import Settings
 from lionagi.utils import UNDEFINED, alcall, bcall, copy
@@ -548,12 +549,38 @@ class Branch(Element, Communicatable, Relational):
 
     def connect(
         self,
-        name: str,
-        imodel: iModel,
-        request_options: type[BaseModel],
+        provider: str = None,
+        base_url: str = None,
+        endpoint: str | EndPoint = "chat",
+        endpoint_params: list[str] | None = None,
+        api_key: str = None,
+        queue_capacity: int = 100,
+        capacity_refresh_time: float = 60,
+        interval: float | None = None,
+        limit_requests: int = None,
+        limit_tokens: int = None,
+        invoke_with_endpoint: bool = False,
+        imodel: iModel = None,
+        name: str = None,
+        request_options: type[BaseModel] = None,
         description: str = None,
         update: bool = False,
     ):
+        if not imodel:
+            imodel = iModel(
+                provider=provider,
+                base_url=base_url,
+                endpoint=endpoint,
+                endpoint_params=endpoint_params,
+                api_key=api_key,
+                queue_capacity=queue_capacity,
+                capacity_refresh_time=capacity_refresh_time,
+                interval=interval,
+                limit_requests=limit_requests,
+                limit_tokens=limit_tokens,
+                invoke_with_endpoint=invoke_with_endpoint,
+            )
+
         if not update and name in self.tools:
             raise ValueError(f"Tool with name '{name}' already exists.")
 
@@ -563,13 +590,13 @@ class Branch(Element, Communicatable, Relational):
             self._log_manager.log(Log.create(api_call))
             return api_call.response
 
-        _connect.__name__ = name
+        _connect.__name__ = name or imodel.endpoint.name
         if description:
             _connect.__doc__ = description
 
         tool = Tool(
             func_callable=_connect,
-            request_options=request_options,
+            request_options=request_options or imodel.request_options,
         )
         self._action_manager.register_tools(tool, update=update)
 
@@ -1521,6 +1548,10 @@ class Branch(Element, Communicatable, Relational):
         self,
         instruct: Instruct | dict[str, Any],
         interpret: bool = False,
+        interpret_domain: str | None = None,
+        interpret_style: str | None = None,
+        interpret_sample: str | None = None,
+        interpret_kwargs: dict | None = None,
         tools: Any = None,
         tool_schemas: Any = None,
         response_format: type[BaseModel] = None,
@@ -1547,6 +1578,14 @@ class Branch(Element, Communicatable, Relational):
             interpret (bool, optional):
                 If `True`, first interprets (`branch.interpret`) the instructions to refine them
                 before proceeding. Defaults to `False`.
+            interpret_domain (str | None, optional):
+                Optional domain hint for the interpretation step.
+            interpret_style (str | None, optional):
+                Optional style hint for the interpretation step.
+            interpret_sample (str | None, optional):
+                Optional sample hint for the interpretation step.
+            interpret_kwargs (dict | None, optional):
+                Additional arguments for the interpretation step.
             tools (Any, optional):
                 Tools to be made available for the ReAct process. If omitted or `None`,
                 and if no `tool_schemas` are provided, it defaults to `True` (all tools).
@@ -1595,6 +1634,10 @@ class Branch(Element, Communicatable, Relational):
             self,
             instruct,
             interpret=interpret,
+            interpret_domain=interpret_domain,
+            interpret_style=interpret_style,
+            interpret_sample=interpret_sample,
+            interpret_kwargs=interpret_kwargs,
             tools=tools,
             tool_schemas=tool_schemas,
             response_format=response_format,

lionagi/version.py

@@ -1 +1 @@
-__version__ = "0.8.4"
+__version__ = "0.8.5"

{lionagi-0.8.4.dist-info → lionagi-0.8.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: lionagi
-Version: 0.8.4
+Version: 0.8.5
 Summary: An Intelligence Operating System.
 Author-email: HaiyangLi <quantocean.li@gmail.com>
 License:            Apache License

{lionagi-0.8.4.dist-info → lionagi-0.8.5.dist-info}/RECORD

@@ -4,7 +4,7 @@ lionagi/_errors.py,sha256=wNKdnVQvE_CHEstK7htrrj334RA_vbGcIds-3pUiRkc,455
 lionagi/_types.py,sha256=9g7iytvSj3UjZxD-jL06_fxuNfgZyWT3Qnp0XYp1wQU,63
 lionagi/settings.py,sha256=k9zRJXv57TveyfHO3Vr9VGiKrSwlRUUVKt5zf6v9RU4,1627
 lionagi/utils.py,sha256=QbF4E1PG-BaRcEVH3kJIYCJVNq-oRNoTxjda5k8NYW4,73177
-lionagi/version.py,sha256=jhHEJFZWhkQDemoZMomBYq-RNrKXknYzUaeIU9A6XsI,22
+lionagi/version.py,sha256=K0kGrhh1kzVisZcoSkeuJdC06rTwxufV05Vy2hOVGoo,22
 lionagi/libs/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
 lionagi/libs/parse.py,sha256=tpEbmIRGuHhLCJlUlm6fjmqm_Z6XJLAXGNFHNuk422I,1011
 lionagi/libs/file/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
@@ -48,7 +48,7 @@ lionagi/operations/__init__.py,sha256=O7nV0tedpUe7_OlUWmCcduGPFtqtzWZcR_SIOnjLsr
 lionagi/operations/manager.py,sha256=H7UY86PIxvxKdzJY9YVsWyJcqlwLWhVyvm4sYePH_uY,565
 lionagi/operations/types.py,sha256=LIa68xcyKLVafof-DSFwKtSkneuYPFqrtGyClohYI6o,704
 lionagi/operations/utils.py,sha256=Twy6L_UFt9JqJFRYuKKTKVZIXsePidNl5ipcYcCbesI,1220
-lionagi/operations/ReAct/ReAct.py,sha256=07F_VhxNnOhOInEZkbY9nT2YiCcnApSibxjz2yAa7Wk,3696
+lionagi/operations/ReAct/ReAct.py,sha256=pmg50Bz859dfoFW4S-fh5-2FUZKkleCVISMp4d6ye5E,4018
 lionagi/operations/ReAct/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
 lionagi/operations/ReAct/utils.py,sha256=yBsbaZm3NNb-LhdjdK3jVmxSYbp0enWzl8d09iv8oSo,1099
 lionagi/operations/_act/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
@@ -161,11 +161,11 @@ lionagi/protocols/messages/templates/instruction_message.jinja2,sha256=L-ptw5OHx
 lionagi/protocols/messages/templates/system_message.jinja2,sha256=JRKJ0aFpYfaXSFouKc_N4unZ35C3yZTOWhIrIdCB5qk,215
 lionagi/protocols/messages/templates/tool_schemas.jinja2,sha256=ozIaSDCRjIAhLyA8VM6S-YqS0w2NcctALSwx4LjDwII,126
 lionagi/service/__init__.py,sha256=DMGXIqPsmut9H5GT0ZeSzQIzYzzPwI-2gLXydpbwiV8,21
-lionagi/service/imodel.py,sha256=Mv5wVsmZj4hYmrvXfHVmA-fskfAKrDY0mbdmgOCAGoE,11946
+lionagi/service/imodel.py,sha256=zQq9cdVPpEAPB7IscntExvtHOYA5ToiWonmD2n93pEw,12273
 lionagi/service/manager.py,sha256=MKSYBkg23s7YhZy5GEFdnpspEnhPVfFhpkpoJe20D7k,1435
 lionagi/service/types.py,sha256=v9SAn5-GTmds4Mar13Or_VFrRHCinBK99dmeDUd-QNk,486
 lionagi/service/endpoints/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
-lionagi/service/endpoints/base.py,sha256=RbBBQIbtNtkP1LT9U3ZrhL35SfmVIpaq-AGW8d538k4,18275
+lionagi/service/endpoints/base.py,sha256=SaYobDBCdKd4398TyFPp5u3PKyMnzEbm6PsoUxVkZDA,18605
 lionagi/service/endpoints/chat_completion.py,sha256=9ltSQaKPH43WdEDW32_-f5x07I9hOU8g-T_PAG-nYsQ,2529
 lionagi/service/endpoints/match_endpoint.py,sha256=hIGYyok1y53FfI6av5NfYMygRIpDWYZbdCj0pJJfmPY,1874
 lionagi/service/endpoints/rate_limited_processor.py,sha256=umri0FofbyBSFdAQBEsviDB5K6N12LkRiXQgSOorGKg,4663
@@ -176,7 +176,7 @@ lionagi/service/providers/anthropic_/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKU
 lionagi/service/providers/anthropic_/messages.py,sha256=PTZZ2VXVMRHWY84YFIzrft9gVrcH2V-NIq_Phi9_-xI,1760
 lionagi/service/providers/exa_/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 lionagi/service/providers/exa_/models.py,sha256=263KP-JSxbxmomNrFeYjB_cebquoMOsCJeWsiKZ0mL4,5420
-lionagi/service/providers/exa_/search.py,sha256=Z9bOxTjPZbtI_qH4fSePYKSXhkc8N3ZFbCHdsml8YSA,1903
+lionagi/service/providers/exa_/search.py,sha256=Z3pyJH8KiWiquJSJw8Rd6D7x43BwTFHb2ESsgSicCk0,1932
 lionagi/service/providers/exa_/types.py,sha256=8ODjXpFajBE9-DGqBJNS--GObwmLSDi667xS84z_AgA,139
 lionagi/service/providers/groq_/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
 lionagi/service/providers/groq_/chat_completions.py,sha256=578NqQYyrIYjIemyL3bagvFGE6ear_w4S1HNlPWA5mg,1343
@@ -185,11 +185,12 @@ lionagi/service/providers/openai_/chat_completions.py,sha256=SfRcEnMTn3MD59YuZCl
 lionagi/service/providers/openrouter_/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
 lionagi/service/providers/openrouter_/chat_completions.py,sha256=MRf4ZbMCgzNIL4gxUZTD-KeFe8JYDn1Fu40Jph3bCH8,1525
 lionagi/service/providers/perplexity_/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
-lionagi/service/providers/perplexity_/chat_completions.py,sha256=SsDbrtXwQsR4Yu2VMU43KfeS86QWI8UTNhDth5lNWNs,1055
+lionagi/service/providers/perplexity_/chat_completions.py,sha256=jhE-KHWRX6yYEeKWLMCKLgK3bQzieSv2viqQWDP8q0Q,1197
+lionagi/service/providers/perplexity_/models.py,sha256=gXH4XGkhZ4aFxvMSDTlHq9Rz1mhu3aTENXAtE-BIr6U,4866
 lionagi/session/__init__.py,sha256=v8vNyJVIVj8_Oz9RJdVe6ZKUQMYTgDh1VQpnr1KdLaw,112
-lionagi/session/branch.py,sha256=ZLItnjW66__85s0brxhq1LZisjDoVS4HBJSHUpFaxsg,63834
+lionagi/session/branch.py,sha256=JFRX_amOH5puYBIFifvXlk6YztUTi8D2Zy-eck0AFn8,65757
 lionagi/session/session.py,sha256=po6C7PnM0iu_ISHUo4PBzzQ61HFOgcsAUfPoO--eLak,8987
-lionagi-0.8.4.dist-info/METADATA,sha256=MHgfVuup5176MSrXlfX2B_tQUaiKNdGAjtITrJVLCWY,22819
-lionagi-0.8.4.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-lionagi-0.8.4.dist-info/licenses/LICENSE,sha256=VXFWsdoN5AAknBCgFqQNgPWYx7OPp-PFEP961zGdOjc,11288
-lionagi-0.8.4.dist-info/RECORD,,
+lionagi-0.8.5.dist-info/METADATA,sha256=nTeNxoGDS7shH8qi3iSb03arU23BDaQ2DgB9_UJvTEQ,22819
+lionagi-0.8.5.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+lionagi-0.8.5.dist-info/licenses/LICENSE,sha256=VXFWsdoN5AAknBCgFqQNgPWYx7OPp-PFEP961zGdOjc,11288
+lionagi-0.8.5.dist-info/RECORD,,