diff --git a/modules/llm.py b/modules/llm.py index 7b59065..fce550e 100644 --- a/modules/llm.py +++ b/modules/llm.py @@ -6,10 +6,8 @@ from modules.log import * # Ollama Client # https://github.com/ollama/ollama/blob/main/docs/faq.md#how-do-i-configure-ollama-server -import ollama # pip install ollama from ollama import Client as OllamaClient from googlesearch import search # pip install googlesearch-python -import chromadb # pip install chromadb # LLM System Variables OllamaClient(host=ollamaHostName) @@ -22,6 +20,10 @@ llmChat_history = {} trap_list_llm = ("ask:", "askai") ragDEV = False +if ragDEV: + import ollama # pip install ollama + import chromadb # pip install chromadb + meshBotAI = """ FROM {llmModel} SYSTEM