lionagi 0.5.5__py3-none-any.whl → 0.6.1__py3-none-any.whl

lionagi/service/providers/openrouter_/chat_completions.py

@@ -0,0 +1,62 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from lionagi.service.endpoints.chat_completion import ChatCompletionEndPoint
+
+CHAT_COMPLETION_CONFIG = {
+    "provider": "openrouter",
+    "base_url": "https://openrouter.ai/api/v1",
+    "endpoint": "chat/completions",
+    "method": "post",
+    "openai_compatible": True,
+    "is_invokeable": True,
+    "requires_tokens": True,
+    "is_streamable": True,
+    "required_kwargs": {
+        "messages",
+        "model",
+    },
+    "deprecated_kwargs": {
+        "max_tokens",
+        "function_call",
+        "functions",
+    },
+    "optional_kwargs": {
+        "store",
+        "reasoning_effort",
+        "metadata",
+        "frequency_penalty",
+        "logit_bias",
+        "logprobs",
+        "top_logprobs",
+        "max_completion_tokens",
+        "n",
+        "modalities",
+        "prediction",
+        "audio",
+        "presence_penalty",
+        "response_format",
+        "seed",
+        "service_tier",
+        "stop",
+        "stream",
+        "stream_options",
+        "temperature",
+        "top_p",
+        "tools",
+        "tool_choice",
+        "parallel_tool_calls",
+        "user",
+    },
+    "allowed_roles": ["user", "assistant", "system"],
+}
+
+
+class OpenRouterChatCompletionEndPoint(ChatCompletionEndPoint):
+    """
+    Documentation: https://openrouter.ai/docs/quick-start#quick-start
+    """
+
+    def __init__(self, config: dict = CHAT_COMPLETION_CONFIG):
+        super().__init__(config)

lionagi/service/providers/perplexity_/__init__.py

@@ -0,0 +1,3 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0

lionagi/service/providers/perplexity_/chat_completions.py

@@ -0,0 +1,40 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from lionagi.service.endpoints.chat_completion import ChatCompletionEndPoint
+
+CHAT_COMPLETION_CONFIG = {
+    "provider": "perplexity",
+    "base_url": "https://api.perplexity.ai",
+    "endpoint": "chat/completions",
+    "method": "post",
+    "openai_compatible": True,
+    "is_invokeable": True,
+    "requires_tokens": True,
+    "is_streamable": True,
+    "required_kwargs": {
+        "messages",
+        "model",
+    },
+    "optional_kwargs": {
+        "max_tokens",
+        "temperature",
+        "top_p",
+        "search_domain_filter",
+        "return_images",
+        "return_related_questions",
+        "search_recency_filter",
+        "top_k",
+        "stream",
+        "presence_penalty",
+        "frequency_penalty",
+    },
+    "allowed_roles": ["user", "assistant"],
+}
+
+
+class PerplexityChatCompletionEndPoint(ChatCompletionEndPoint):
+
+    def __init__(self, config: dict = CHAT_COMPLETION_CONFIG):
+        super().__init__(config)

lionagi/service/types.py

@@ -0,0 +1,18 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0
+
+from .endpoints.base import APICalling, EndPoint
+from .endpoints.rate_limited_processor import RateLimitedAPIExecutor
+from .endpoints.token_calculator import TokenCalculator
+from .imodel import iModel
+from .manager import iModelManager
+
+__all__ = (
+    "APICalling",
+    "EndPoint",
+    "RateLimitedAPIExecutor",
+    "TokenCalculator",
+    "iModel",
+    "iModelManager",
+)

lionagi/session/__init__.py

@@ -0,0 +1,3 @@
+# Copyright (c) 2023 - 2024, HaiyangLi <quantocean.li at gmail dot com>
+#
+# SPDX-License-Identifier: Apache-2.0