Update factory.py (#1657)

This commit is contained in:
krescent
2024-08-08 15:06:15 +05:30
committed by GitHub
parent e190445492
commit 8c8f4120f4
+1 -1
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",