Add PremAI LLM implementation

This commit is contained in:
Alex
2024-02-13 14:08:55 +00:00
parent 2c991583ff
commit b7927d8d75
3 changed files with 39 additions and 1 deletions

View File

@@ -4,6 +4,7 @@ from application.llm.huggingface import HuggingFaceLLM
from application.llm.llama_cpp import LlamaCpp
from application.llm.anthropic import AnthropicLLM
from application.llm.docsgpt_provider import DocsGPTAPILLM
from application.llm.premai import PremAILLM
@@ -15,7 +16,8 @@ class LLMCreator:
'huggingface': HuggingFaceLLM,
'llama.cpp': LlamaCpp,
'anthropic': AnthropicLLM,
'docsgpt': DocsGPTAPILLM
'docsgpt': DocsGPTAPILLM,
'premai': PremAILLM,
}
@classmethod