From d862e3c7be9605614e0ff380fbd5fc672ac7230e Mon Sep 17 00:00:00 2001
From: Ivar Flakstad <69173633+ivarflakstad@users.noreply.github.com>
Date: Tue, 9 Apr 2024 11:52:42 +0200
Subject: [PATCH] make fmt

---
 backend/app/database/redis.py               |  1 -
 backend/app/settings.py                     |  7 ++++---
 backend/prompt_experimentation/pe_script.py | 10 ++++------
 3 files changed, 8 insertions(+), 10 deletions(-)

diff --git a/backend/app/database/redis.py b/backend/app/database/redis.py
index 75d5bd0d..27afc37b 100644
--- a/backend/app/database/redis.py
+++ b/backend/app/database/redis.py
@@ -1,5 +1,4 @@
 import logging
-import random
 
 import redis.asyncio as aioredis
 
diff --git a/backend/app/settings.py b/backend/app/settings.py
index a2061c5c..67adf707 100644
--- a/backend/app/settings.py
+++ b/backend/app/settings.py
@@ -1,5 +1,4 @@
 # Load environment variables from .env file
-import pydantic
 from pydantic import SecretStr
 from pydantic_settings import BaseSettings, SettingsConfigDict
 
@@ -142,8 +141,10 @@ class AIModelsSettings(BaseSettings):
 
 class Settings(BaseSettings):
     model_config = SettingsConfigDict(
-        extra='allow',
-        env_file=".env", env_file_encoding="utf-8", env_nested_delimiter="__"
+        extra="allow",
+        env_file=".env",
+        env_file_encoding="utf-8",
+        env_nested_delimiter="__",
     )
 
     postgres_engine: SecretStr
diff --git a/backend/prompt_experimentation/pe_script.py b/backend/prompt_experimentation/pe_script.py
index e98a4dbf..c20d4675 100644
--- a/backend/prompt_experimentation/pe_script.py
+++ b/backend/prompt_experimentation/pe_script.py
@@ -6,8 +6,8 @@
 import wandb
 from dotenv import load_dotenv
 from dspy.teleprompt import BootstrapFewShot
-from pydantic import SecretStr
 from tqdm import tqdm
+
 from app.settings import Settings
 
 settings = Settings()
@@ -70,8 +70,7 @@
     "Communication and Education, Synthetic Biology, Systems Biology, Zoology": "2",
     # bioarxiv
     "3. useful only for retrieving the drug related information like molecular "
-    "weights,similarities,smile codes, target medicines, effects on other medicine":
-        "3",
+    "weights,similarities,smile codes, target medicines, effects on other medicine": "3",
     # chembl
 }
 
@@ -94,8 +93,7 @@
 
 if args.llm == "gpt-3.5-turbo":
     turbo = dspy.OpenAI(
-        model="gpt-3.5-turbo",
-        api_key=settings.openai.api_key.get_secret_value()
+        model="gpt-3.5-turbo", api_key=settings.openai.api_key.get_secret_value()
     )
 elif args.llm == "gemma-7b-it":
     turbo = dspy.GROQ(
@@ -202,7 +200,7 @@ def metric(gold, pred, trace=None):
 teleprompter = BootstrapFewShot(metric=metric)
 compiled_rag = teleprompter.compile(Router_module(), trainset=total_data[:DEV_NUM])
 
-for x in tqdm(total_data[DEV_NUM: DEV_NUM + 20]):
+for x in tqdm(total_data[DEV_NUM : DEV_NUM + 20]):
     pred = compiled_rag.generate_answer(question=x.question)
     print(f"Question: {x.question}")
     print(f"Predicted Answer: {pred.answer}")