Linting

StacklokLabs · Nov 25, 2024 · 1ee0819 · 1ee0819
1 parent 84a0aeb
commit 1ee0819
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 5 deletions.
diff --git a/promptwright/config.py b/promptwright/config.py
@@ -1,6 +1,5 @@
 """Configuration handling for YAML-based promptwright configurations."""
 
-import os
 from dataclasses import dataclass
 
 import yaml

diff --git a/promptwright/engine.py b/promptwright/engine.py
@@ -207,10 +207,6 @@ def create_data(  # noqa: PLR0912
                                 ],
                                 "temperature": self.args.temperature,
                             }
-
-                            # # Add base_url only for Ollama
-                            # if self.model_name.startswith("ollama/"):
-                            #     completion_args["base_url"] = "http://localhost:11434"
 
                             responses = litellm.batch_completion(**completion_args)