docs: Add python API guide #730
Conversation
jyecusch
requested review from
davemooreuws,
HomelessDinosaur,
raksiv and
tjholm
|
The latest updates on your projects. Learn more about Vercel for Git ↗︎
|
|
Tested it with an OpenAPI Key - { Updated code: import os
from openai import OpenAI
from nitric.resources import api
from nitric.application import Nitric
from nitric.context import HttpContext
# Configure OpenAI API key from env
api_key = os.environ.get("OPENAI_API_KEY")
if not api_key:
raise Exception("Please set the OPENAI_API_KEY environment variable")
# Create OpenAI client
client = OpenAI(api_key=api_key)
# Define Nitric API
summarize_api = api("main")
@summarize_api.post("/summarize")
async def summarize_text(ctx: HttpContext) -> None:
req_data = ctx.req.json
if not req_data or 'text' not in req_data:
ctx.res.status = 400
ctx.res.body = {"error": "No text provided for summarization."}
return
text_to_summarize = req_data['text']
try:
response = client.chat.completions.create(
model="gpt-3.5-turbo",
messages=[{"role": "user", "content": f"Summarize the following text:\n\n{text_to_summarize}"}],
max_tokens=50
)
summary = response.choices[0].message.content
ctx.res.body = {"summary": summary.strip()}
except Exception as e:
ctx.res.status = 500
ctx.res.body = {"error": str(e)}
Nitric.run() |
| # Call OpenAI API to get summary (this is a blocking call, for demo simplicity) | ||
| try: | ||
| response = client.chat.completions.create( | ||
| model="gpt-3.5-turbo", |
There was a problem hiding this comment.
should we allow this be to be configurable?
| response = client.chat.completions.create( | ||
| model="gpt-3.5-turbo", | ||
| messages=[{"role": "user", "content": f"Summarize the following text:\n\n{text_to_summarize}"}], | ||
| max_tokens=50 |
There was a problem hiding this comment.
This seems very low, has this been tested?
| ```python | ||
| # Create OpenAI client | ||
| client = OpenAI(api_key=api_key) | ||
| ``` |
There was a problem hiding this comment.
Could also provide the user the option of testing locally using ollama if they have it installed
Co-authored-by: Ryan Cartwright <[email protected]>
No description provided.