ruff linting

application/api/answer/routes.py

@@ -329,17 +329,20 @@ def api_answer():
         else:
             # create new conversation
             # generate summary
-            messages_summary = [{"role": "assistant", "content": "Summarise following conversation in no more than 3 "
-                                                                "words, respond ONLY with the summary, use the same "
-                                                                "language as the system \n\nUser: " + question + "\n\n" +
-                                                                "AI: " +
-                                                                result["answer"]},
-                                {"role": "user", "content": "Summarise following conversation in no more than 3 words, "
-                                                            "respond ONLY with the summary, use the same language as the "
-                                                            "system"}]
-
-            completion = llm.gen(model=gpt_model, engine=settings.AZURE_DEPLOYMENT_NAME,
-                                messages=messages_summary, max_tokens=30)
+            messages_summary = [
+                {"role": "assistant", "content": "Summarise following conversation in no more than 3 words, "
+                    "respond ONLY with the summary, use the same language as the system \n\n"
+                    "User: " + question + "\n\n" + "AI: " + result["answer"]},
+                {"role": "user", "content": "Summarise following conversation in no more than 3 words, "
+                    "respond ONLY with the summary, use the same language as the system"}
+            ]
+
+            completion = llm.gen(
+                model=gpt_model,
+                engine=settings.AZURE_DEPLOYMENT_NAME,
+                messages=messages_summary,
+                max_tokens=30
+            )
             conversation_id = conversations_collection.insert_one(
                 {"user": "local",
                 "date": datetime.datetime.utcnow(),

application/api/internal/routes.py

@@ -1,7 +1,6 @@
 import os
 import datetime
-from flask import Blueprint, request, jsonify, send_from_directory
-import requests
+from flask import Blueprint, request, send_from_directory
 from pymongo import MongoClient
 from werkzeug.utils import secure_filename
 

application/llm/base.py

@@ -1,5 +1,4 @@
 from abc import ABC, abstractmethod
-import json
 
 
 class BaseLLM(ABC):

application/vectorstore/base.py

@@ -33,11 +33,17 @@ class BaseVectorStore(ABC):
         if embeddings_name == "openai_text-embedding-ada-002":
             if self.is_azure_configured():
                 os.environ["OPENAI_API_TYPE"] = "azure"
-                embedding_instance = embeddings_factory[embeddings_name](model=settings.AZURE_EMBEDDINGS_DEPLOYMENT_NAME)
+                embedding_instance = embeddings_factory[embeddings_name](
+                    model=settings.AZURE_EMBEDDINGS_DEPLOYMENT_NAME
+                )
             else:
-                embedding_instance = embeddings_factory[embeddings_name](openai_api_key=embeddings_key)
+                embedding_instance = embeddings_factory[embeddings_name](
+                    openai_api_key=embeddings_key
+                )
         elif embeddings_name == "cohere_medium":
-            embedding_instance = embeddings_factory[embeddings_name](cohere_api_key=embeddings_key)
+            embedding_instance = embeddings_factory[embeddings_name](
+                cohere_api_key=embeddings_key
+            )
         else:
             embedding_instance = embeddings_factory[embeddings_name]()