Fortsatta ändringar för tester och felsökningsändamål

This commit is contained in:
2024-07-17 09:21:34 +02:00
parent 1b7142f157
commit cecc2f7520
2 changed files with 13 additions and 11 deletions
+5 -3
View File
@@ -2,20 +2,22 @@
# Import the necessary functions from ollama, Flask, requests, threading # Import the necessary functions from ollama, Flask, requests, threading
from ollama import Client from ollama import Client
from flask import Flask, request, jsonify from flask import Flask, request, jsonify
import requests #import requests
#import threading #import threading
# Initialize a Flask application # Initialize a Flask application
app = Flask(__name__) app = Flask(__name__)
@app.route('/chat', methods=['POST']) @app.route('/api/chat', methods=['POST'])
def chat(): def chat():
if request.method == 'POST':
# Get the message from the JSON in the request body # Get the message from the JSON in the request body
data = request.get_json() data = request.get_json()
message = data.get('query') message = data.get('query')
if message: if message:
response = requests.post('http://localhost:11434', json={"model": "llama3",'prompt': message}) # response = requests.post('http://localhost:11434', json={"model": "llama3",'prompt': message})
response = request.post('http://localhost:11434', json={"model": "llama3",'prompt': message})
return jsonify({'response': response.json().get('result')}) return jsonify({'response': response.json().get('result')})
else: else:
return jsonify({'error': 'No query provided'}), 400 return jsonify({'error': 'No query provided'}), 400
+1 -1
View File
@@ -41,7 +41,7 @@ function sendMessage() {
const xhr = new XMLHttpRequest(); const xhr = new XMLHttpRequest();
// Configure the request // Configure the request
xhr.open('POST', '/chat'); // Set method to POST and URL xhr.open('POST', '/api/chat'); // Set method to POST and URL
xhr.setRequestHeader('Content-Type', 'application/json'); // Essential for sending JSON xhr.setRequestHeader('Content-Type', 'application/json'); // Essential for sending JSON
// Handle response data // Handle response data