From 38bfcc1581e3376f98736549bb86b370c23f5282 Mon Sep 17 00:00:00 2001 From: SpudGunMan Date: Tue, 15 Oct 2024 19:33:04 -0700 Subject: [PATCH] Update llm.py --- modules/llm.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/modules/llm.py b/modules/llm.py index fce550e..145e09a 100644 --- a/modules/llm.py +++ b/modules/llm.py @@ -9,6 +9,12 @@ from modules.log import * from ollama import Client as OllamaClient from googlesearch import search # pip install googlesearch-python +ragDEV = False + +if ragDEV: + import ollama # pip install ollama + import chromadb # pip install chromadb + # LLM System Variables OllamaClient(host=ollamaHostName) ollamaClient = OllamaClient() @@ -18,11 +24,7 @@ googleSearchResults = 3 # number of google search results to include in the cont antiFloodLLM = [] llmChat_history = {} trap_list_llm = ("ask:", "askai") -ragDEV = False -if ragDEV: - import ollama # pip install ollama - import chromadb # pip install chromadb meshBotAI = """ FROM {llmModel}