Include OpenAI LLM in quickstart example

parkervg · Oct 16, 2024 · c8e3058 · c8e3058
1 parent b481118
commit c8e3058
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -157,9 +157,10 @@ import pandas as pd
 
 from blendsql import blend, LLMMap, LLMQA, LLMJoin
 from blendsql.db import Pandas
-from blendsql.models import TransformersLLM
+from blendsql.models import TransformersLLM, OpenaiLLM
 
 # Load model
+# model = OpenaiLLM("gpt-4o") # If you have a .env present with OpenAI API keys
 model = TransformersLLM('Qwen/Qwen1.5-0.5B')
 
 # Prepare our local database

diff --git a/docs/quickstart.md b/docs/quickstart.md
@@ -9,9 +9,10 @@ import pandas as pd
 
 from blendsql import blend, LLMMap, LLMQA, LLMJoin
 from blendsql.db import Pandas
-from blendsql.models import TransformersLLM
+from blendsql.models import TransformersLLM, OpenaiLLM
 
 # Load model
+# model = OpenaiLLM("gpt-4o") # If you have a .env present with OpenAI API keys
 model = TransformersLLM('Qwen/Qwen1.5-0.5B')
 
 # Prepare our local database