TashaSkyUp
40831eaa28
- Add detailed docstrings to AgentNode, BasicRecursionFilterNode, and DocumentChunkRecursionFilterNode for clarity and documentation. - Improve input validation and error handling in agent nodes. - Update .gitignore to cover more environment, config, and IDE files. - Refactor LiteLLMCustomEndpointProvider for custom API endpoints. - Add conditional imports and dummy functions for LightRAG when unavailable. - Update config.yaml to use environment variables for API keys and clarify usage. - Refine requirements.txt to avoid heavy LightRAG dependencies. - Add tests/__init__.py with test suite metadata; remove redundant test files. - Fix CustomDict import and model config handling in litellmnodes.py. - Improve chunk processing and recursion filter logic in agent nodes. - Update utils/__init__.py to include env_config.
25 lines
615 B
Python
Executable File
25 lines
615 B
Python
Executable File
from .base import (
|
|
AgentBaseNode,
|
|
)
|
|
from .nodes import (
|
|
AgentNode,
|
|
BasicRecursionFilterNode,
|
|
DocumentChunkRecursionFilterNode,
|
|
)
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"AgentNode": AgentNode,
|
|
"BasicRecursionFilterNode": BasicRecursionFilterNode,
|
|
"DocumentChunkRecursionFilterNode": DocumentChunkRecursionFilterNode,
|
|
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"Iterative Completion Agent": "AgentNode",
|
|
"Completion Enhancement Filter": "BasicRecursionFilterNode",
|
|
"Document Chunk Processor": "DocumentChunkRecursionFilterNode",
|
|
|
|
}
|
|
|
|
__all__ = ['NODE_CLASS_MAPPINGS', 'NODE_DISPLAY_NAME_MAPPINGS']
|