Commit Graph

6 Commits

Author SHA1 Message Date
Lorenze Jay
902c330113 Enhance CustomLLM and JWTAuthLLM initialization with model parameter
- Update CustomLLM to accept a model parameter during initialization
- Modify test cases to include the new model argument
- Ensure JWTAuthLLM and TimeoutHandlingLLM also utilize the model parameter in their constructors
- Update type hints in create_llm function to support both LLM and BaseLLM types
2025-03-12 08:16:59 -07:00
Lorenze Jay
709941c4c7 Refactor LLM module by extracting BaseLLM to a separate file
This commit moves the BaseLLM abstract base class from llm.py to a new file llms/base_llm.py to improve code organization. The changes include:

- Creating a new file src/crewai/llms/base_llm.py
- Moving the BaseLLM class to the new file
- Updating imports in __init__.py and llm.py to reflect the new location
- Updating test cases to use the new import path

The refactoring maintains the existing functionality while improving the project's module structure.
2025-03-04 15:54:46 -08:00
Devin AI
963ed23b63 Enhance custom LLM implementation with better error handling, documentation, and test coverage
Co-Authored-By: Joe Moura <joao@crewai.com>
2025-03-04 17:50:52 +00:00
Devin AI
7201161207 Fix linting issues with import sorting
Co-Authored-By: Joe Moura <joao@crewai.com>
2025-03-04 17:19:36 +00:00
Devin AI
687303ad63 Fix import sorting and type annotations
Co-Authored-By: Joe Moura <joao@crewai.com>
2025-03-04 17:13:12 +00:00
Devin AI
ec8e705bbc Add support for custom LLM implementations
Co-Authored-By: Joe Moura <joao@crewai.com>
2025-03-04 17:09:17 +00:00