Hämtar url_server och model när frågor skickas till api/chat

2024-07-28 23:24:39 +02:00
parent 1b5624018c
commit d797440288
1 changed files with 2 additions and 1 deletions
@@ -67,7 +67,8 @@ def chat(url_server = "http://localhost:11434/api/generate", model = "phi3:mini"
    # Get the message from the JSON in the request body
    data = request.get_json()
    message = data.get('query')
-
+    url_server = data.get('url_server', url_server)  # Use provided URL or default
+    model = data.get('model', model)  # Use provided model or default
    logger.debug("data = %s\nmessage = %s", str(data), str(message))
    try:
        url = url_server