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.
21 lines
595 B
Python
Executable File
21 lines
595 B
Python
Executable File
"""
|
|
ComfyUI_LiteLLM Test Suite
|
|
|
|
This package contains tests for the ComfyUI_LiteLLM agent nodes and functionality.
|
|
|
|
Test files:
|
|
- smoke_test.py: Basic smoke tests
|
|
- test_simple_real.py: Comprehensive agent improvement tests
|
|
- test_kluster_real.py: Real API integration tests
|
|
- test_hash_utils.py: Utility function tests
|
|
- test_agent_structure.py: Code structure and documentation tests
|
|
"""
|
|
|
|
__version__ = "1.0.0"
|
|
__author__ = "ComfyUI_LiteLLM Team"
|
|
|
|
# Test configuration
|
|
TEST_TIMEOUT = 30 # seconds
|
|
API_ENDPOINT = "https://api.kluster.ai/v1"
|
|
TEST_MODEL = "mistralai/Mistral-Nemo-Instruct-2407"
|