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:
provider_to_class = {
"ollama": "mem0.llms.ollama.py.OllamaLLM",
"ollama": "mem0.llms.ollama.OllamaLLM",
"openai": "mem0.llms.openai.OpenAILLM",
"groq": "mem0.llms.groq.GroqLLM",
"together": "mem0.llms.together.TogetherLLM",