Optimize Ollama LLM driver

This commit is contained in:
yangdx
2025-05-14 01:13:03 +08:00
parent aa36894d6e
commit b836d02cac
3 changed files with 75 additions and 41 deletions

View File

@@ -447,7 +447,7 @@ rag = LightRAG(
embedding_func=EmbeddingFunc(
embedding_dim=768,
max_token_size=8192,
func=lambda texts: ollama_embedding(
func=lambda texts: ollama_embed(
texts,
embed_model="nomic-embed-text"
)