blob: b11a12b455e8fe2b5c0f7b5d29168929c962f22c (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
from google import genai
import json
import requests
import Constants
import AIParams
client = genai.Client(api_key="AIzaSyAdB7yo0qcnwHeC4T2rRaSXD588JRw94oQ")
def run_ai_query(url):
req_url = f"https://r.jina.ai/{url}"
res = requests.get(req_url)
markdown_content = res.text
prompt = AIParams.PROMPT.format(schema=AIParams.RECIPE_SCHEMA, markdown=markdown_content)
ai_res = client.models.generate_content(model="gemini-2.0-flash", contents=prompt)
cleaned_text = ai_res.text.strip("```").strip("```json")
recipe_json = json.loads(cleaned_text)
return {"success": True, "data": recipe_json}
|
