From 84a0aeb00f2a223501d10ec831c5c93f5fed01da Mon Sep 17 00:00:00 2001
From: Luke Hinds <luke@stacklok.com>
Date: Mon, 25 Nov 2024 14:36:34 +0000
Subject: [PATCH] Clean up

---
 README.md              | 8 ++++++++
 promptwright/config.py | 3 ---
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index 5e7b74a..7508d93 100644
--- 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
index e7eb9b7..d15e2e5 100644
--- 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}"