Spaces:
Sleeping
Sleeping
fix decimal values
Browse files
app.py
CHANGED
@@ -202,7 +202,7 @@ with gr.Blocks(css=css) as demo:
|
|
202 |
prompt=prompt,
|
203 |
)}]
|
204 |
response_format = {"type": "json", "value": {"properties": {"data": {"type": "array", "items": format, "minItems": len(rows), "maxItems": len(rows)}}, "required": ["data"]}}
|
205 |
-
yield from ijson.items(StringIteratorIO(stream_reponse(messages, response_format=response_format)), "data.item", buf_size=4)
|
206 |
|
207 |
|
208 |
def stream_rewrite_dataset_row_by_row(dataset: str, rows: list[dict[str, str]], prompt: str, format: str, input_preview_rows: list[dict[str, str]], output_preview_rows: list[dict[str, str]]) -> Iterator[dict[str, str]]:
|
@@ -216,7 +216,7 @@ with gr.Blocks(css=css) as demo:
|
|
216 |
)}]
|
217 |
response_format = {"type": "json", "value": {"properties": {"data": {"type": "array", "items": format, "minItems": len(rows), "maxItems": len(rows)}}, "required": ["data"]}}
|
218 |
try:
|
219 |
-
yield from ijson.items(StringIteratorIO(stream_reponse(messages, response_format=response_format)), "data.item", buf_size=4)
|
220 |
except ijson.IncompleteJSONError as e:
|
221 |
print(f"{type(e).__name__}: {e}")
|
222 |
print("Warning: Some rows were missing during ReWriting.")
|
|
|
202 |
prompt=prompt,
|
203 |
)}]
|
204 |
response_format = {"type": "json", "value": {"properties": {"data": {"type": "array", "items": format, "minItems": len(rows), "maxItems": len(rows)}}, "required": ["data"]}}
|
205 |
+
yield from ijson.items(StringIteratorIO(stream_reponse(messages, response_format=response_format)), "data.item", buf_size=4, use_float=True)
|
206 |
|
207 |
|
208 |
def stream_rewrite_dataset_row_by_row(dataset: str, rows: list[dict[str, str]], prompt: str, format: str, input_preview_rows: list[dict[str, str]], output_preview_rows: list[dict[str, str]]) -> Iterator[dict[str, str]]:
|
|
|
216 |
)}]
|
217 |
response_format = {"type": "json", "value": {"properties": {"data": {"type": "array", "items": format, "minItems": len(rows), "maxItems": len(rows)}}, "required": ["data"]}}
|
218 |
try:
|
219 |
+
yield from ijson.items(StringIteratorIO(stream_reponse(messages, response_format=response_format)), "data.item", buf_size=4, use_float=True)
|
220 |
except ijson.IncompleteJSONError as e:
|
221 |
print(f"{type(e).__name__}: {e}")
|
222 |
print("Warning: Some rows were missing during ReWriting.")
|