update text-to-sql cookbook

This commit is contained in:
zachary62
2025-05-31 21:10:29 -04:00
parent 2334db7e71
commit 6cf3ec4984
4 changed files with 119 additions and 2 deletions
@@ -0,0 +1,14 @@
import os
from openai import OpenAI
def call_llm(prompt):
client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY", "your-api-key"))
r = client.chat.completions.create(
model="gpt-4o",
messages=[{"role": "user", "content": prompt}]
)
return r.choices[0].message.content
# Example usage
if __name__ == "__main__":
print(call_llm("Tell me a short joke"))