Skip to content

Commit 4d5975d

Browse files
author
SentienceDEV
committed
fix async mismatch
1 parent 6388352 commit 4d5975d

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

sentience/read.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -223,7 +223,7 @@ def extract(
223223
schema_desc = json.dumps(schema.model_json_schema(), ensure_ascii=False)
224224
system = "You extract structured data from markdown content. " "Return only JSON. No prose."
225225
user = f"QUERY:\n{query}\n\nSCHEMA:\n{schema_desc}\n\nCONTENT:\n{content}"
226-
response = await llm.generate_async(system, user)
226+
response = llm.generate(system, user)
227227
raw = response.content.strip()
228228

229229
if schema is None:
@@ -257,7 +257,7 @@ async def extract_async(
257257
schema_desc = json.dumps(schema.model_json_schema(), ensure_ascii=False)
258258
system = "You extract structured data from markdown content. " "Return only JSON. No prose."
259259
user = f"QUERY:\n{query}\n\nSCHEMA:\n{schema_desc}\n\nCONTENT:\n{content}"
260-
response = llm.generate(system, user)
260+
response = await llm.generate_async(system, user)
261261
raw = response.content.strip()
262262

263263
if schema is None:

0 commit comments

Comments
 (0)