Update factory.py (#1657)

This commit is contained in:
krescent
2024-08-08 17:36:15 +08:00
committed by GitHub
parent e190445492
commit 8c8f4120f4

View File

@@ -11,7 +11,7 @@ def load_class(class_type):
class LlmFactory: class LlmFactory:
provider_to_class = { provider_to_class = {
"ollama": "mem0.llms.ollama.py.OllamaLLM", "ollama": "mem0.llms.ollama.OllamaLLM",
"openai": "mem0.llms.openai.OpenAILLM", "openai": "mem0.llms.openai.OpenAILLM",
"groq": "mem0.llms.groq.GroqLLM", "groq": "mem0.llms.groq.GroqLLM",
"together": "mem0.llms.together.TogetherLLM", "together": "mem0.llms.together.TogetherLLM",