From 74cdf050cbbada1705f3b0eb69c5e9fa2d3eb506 Mon Sep 17 00:00:00 2001 From: Asuka Minato Date: Fri, 18 Jul 2025 21:06:25 +0900 Subject: [PATCH] fmt --- api/core/rag/datasource/vdb/pgvecto_rs/pgvecto_rs.py | 2 +- api/models/dataset.py | 3 +-- 2 files changed, 2 insertions(+), 3 deletions(-) diff --git a/api/core/rag/datasource/vdb/pgvecto_rs/pgvecto_rs.py b/api/core/rag/datasource/vdb/pgvecto_rs/pgvecto_rs.py index 1b567415e6..b0f0eeca38 100644 --- a/api/core/rag/datasource/vdb/pgvecto_rs/pgvecto_rs.py +++ b/api/core/rag/datasource/vdb/pgvecto_rs/pgvecto_rs.py @@ -6,7 +6,7 @@ from uuid import UUID, uuid4 from numpy import ndarray from pgvecto_rs.sqlalchemy import VECTOR # type: ignore from pydantic import BaseModel, model_validator -from sqlalchemy import Float, String, create_engine, insert, select, text +from sqlalchemy import Float, create_engine, insert, select, text from sqlalchemy import text as sql_text from sqlalchemy.dialects import postgresql from sqlalchemy.orm import Mapped, Session, mapped_column diff --git a/api/models/dataset.py b/api/models/dataset.py index 53cc85016e..e5d67a721e 100644 --- a/api/models/dataset.py +++ b/api/models/dataset.py @@ -1,4 +1,3 @@ -from typing import Optional import base64 import enum import hashlib @@ -10,7 +9,7 @@ import pickle import re import time from json import JSONDecodeError -from typing import Any, cast +from typing import Any, Optional, cast from sqlalchemy import func from sqlalchemy.dialects.postgresql import JSONB