|
|
|
@ -132,7 +132,6 @@ class LLM: |
|
|
|
|
|
|
|
|
|
|
|
def _prepare_messages_and_model(self, query, user_input, context, messages, images, model): |
|
|
|
def _prepare_messages_and_model(self, query, user_input, context, messages, images, model): |
|
|
|
"""Prepare messages and select the appropriate model, handling images if present.""" |
|
|
|
"""Prepare messages and select the appropriate model, handling images if present.""" |
|
|
|
print_red('model', model) |
|
|
|
|
|
|
|
if messages: |
|
|
|
if messages: |
|
|
|
messages = [ |
|
|
|
messages = [ |
|
|
|
{"role": i["role"], "content": re.sub(r"\s*\n\s*", "\n", i["content"])} |
|
|
|
{"role": i["role"], "content": re.sub(r"\s*\n\s*", "\n", i["content"])} |
|
|
|
@ -153,7 +152,6 @@ class LLM: |
|
|
|
model = self.get_model(model) |
|
|
|
model = self.get_model(model) |
|
|
|
|
|
|
|
|
|
|
|
self.messages.append(message) |
|
|
|
self.messages.append(message) |
|
|
|
print_red('return model',model) |
|
|
|
|
|
|
|
return model |
|
|
|
return model |
|
|
|
|
|
|
|
|
|
|
|
def _build_headers(self, model, tools, think): |
|
|
|
def _build_headers(self, model, tools, think): |
|
|
|
|