langroid 0.33.6__py3-none-any.whl → 0.33.8__py3-none-any.whl

langroid/__init__.py

@@ -0,0 +1,106 @@
+"""
+Main langroid package
+"""
+
+from . import mytypes
+from . import utils
+
+from . import parsing
+from . import prompts
+from . import cachedb
+
+from . import language_models
+from . import embedding_models
+
+from . import vector_store
+from . import agent
+
+from .agent.base import (
+    Agent,
+    AgentConfig,
+)
+
+from .agent.batch import (
+    run_batch_tasks,
+    llm_response_batch,
+    agent_response_batch,
+)
+
+from .agent.chat_document import (
+    StatusCode,
+    ChatDocument,
+    ChatDocMetaData,
+)
+
+from .agent.tool_message import (
+    ToolMessage,
+)
+
+from .agent.chat_agent import (
+    ChatAgent,
+    ChatAgentConfig,
+)
+
+from .agent.task import Task, TaskConfig
+
+
+from .mytypes import (
+    DocMetaData,
+    Document,
+    Entity,
+)
+
+from .exceptions import InfiniteLoopException
+from .exceptions import LangroidImportError
+
+__all__ = [
+    "mytypes",
+    "exceptions",
+    "utils",
+    "parsing",
+    "prompts",
+    "cachedb",
+    "language_models",
+    "embedding_models",
+    "vector_store",
+    "agent",
+    "Agent",
+    "AgentConfig",
+    "ChatAgent",
+    "ChatAgentConfig",
+    "StatusCode",
+    "ChatDocument",
+    "ChatDocMetaData",
+    "Task",
+    "TaskConfig",
+    "DocMetaData",
+    "Document",
+    "Entity",
+    "ToolMessage",
+    "run_batch_tasks",
+    "llm_response_batch",
+    "agent_response_batch",
+    "InfiniteLoopException",
+    "LangroidImportError",
+]
+
+
+try:
+    from .agent.callbacks.chainlit import (
+        ChainlitAgentCallbacks,
+        ChainlitTaskCallbacks,
+        ChainlitCallbackConfig,
+    )
+
+    ChainlitAgentCallbacks
+    ChainlitTaskCallbacks
+    ChainlitCallbackConfig
+    __all__.extend(
+        [
+            "ChainlitAgentCallbacks",
+            "ChainlitTaskCallbacks",
+            "ChainlitCallbackConfig",
+        ]
+    )
+except ImportError:
+    pass

langroid/agent/__init__.py

@@ -0,0 +1,41 @@
+from .base import Agent, AgentConfig
+from .chat_document import (
+    ChatDocAttachment,
+    ChatDocMetaData,
+    ChatDocLoggerFields,
+    ChatDocument,
+)
+from .chat_agent import ChatAgentConfig, ChatAgent
+from .tool_message import ToolMessage
+from .task import Task
+
+from . import base
+from . import chat_document
+from . import chat_agent
+from . import task
+from . import batch
+from . import tool_message
+from . import tools
+from . import special
+
+
+__all__ = [
+    "Agent",
+    "AgentConfig",
+    "ChatDocAttachment",
+    "ChatDocMetaData",
+    "ChatDocLoggerFields",
+    "ChatDocument",
+    "ChatAgent",
+    "ChatAgentConfig",
+    "ToolMessage",
+    "Task",
+    "base",
+    "chat_document",
+    "chat_agent",
+    "task",
+    "batch",
+    "tool_message",
+    "tools",
+    "special",
+]