Refactor Model class to remove unused API key parameter

Sedrowow · Sep 16, 2024 · e2e8cc7 · e2e8cc7
1 parent d225b5b
commit e2e8cc7
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/app/models/model.py b/app/models/model.py
@@ -3,14 +3,11 @@
 
 
 class Model:
-    def __init__(self, model_name, base_url, api_key, context):
+    def __init__(self, model_name, base_url, context):
         self.model_name = model_name
         self.base_url = base_url
-        self.api_key = api_key
         self.context = context
 
-        if api_key:
-            os.environ['OPENAI_API_KEY'] = api_key
 
     def get_instructions_for_objective(self, *args) -> dict[str, Any]:
         pass