Clean up

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

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -169,6 +169,14 @@ provider: "gemini"  # LLM provider
     model: "gemini-pro"  # Model name
 ```
 
+For Ollama, you would specify the provider as `ollama` and the model as `mistral`
+and so on.
+
+```yaml
+provider: "ollama"  # LLM provider
+    model: "mistral:latest"  # Model name
+```
+
 ##### API Keys
 
 You can set the API key for the provider in the environment variable. The key

diff --git a/promptwright/config.py b/promptwright/config.py
@@ -11,9 +11,6 @@
 
 def construct_model_string(provider: str, model: str) -> str:
     """Construct the full model string for LiteLLM."""
-    # if provider == "openai":
-    #     if "OPENAI_API_KEY" not in os.environ:
-    #         raise ValueError("OPENAI_API_KEY environment variable is required for OpenAI provider")
     return f"{provider}/{model}"