Léo Bourrel commited on
Commit
244cbec
·
1 Parent(s): 7932f4a

clean: isort

Browse files
Files changed (1) hide show
  1. custom_pgvector.py +2 -7
custom_pgvector.py CHANGED
@@ -1,25 +1,20 @@
1
  from __future__ import annotations
2
 
3
- import asyncio
4
  import contextlib
5
  import enum
6
  import json
7
  import logging
8
- from functools import partial
9
- from typing import (Any, Callable, Dict, Generator, Iterable, List, Optional,
10
- Tuple, Type)
11
 
12
- import numpy as np
13
  import pandas as pd
14
  import sqlalchemy
15
  from langchain.docstore.document import Document
16
  from langchain.schema.embeddings import Embeddings
17
  from langchain.utils import get_from_dict_or_env
18
  from langchain.vectorstores.base import VectorStore
19
- from langchain.vectorstores.pgvector import BaseModel
20
  from pgvector.sqlalchemy import Vector
21
  from sqlalchemy import delete, text
22
- from sqlalchemy.orm import Session, declarative_base, relationship
23
 
24
 
25
  class DistanceStrategy(str, enum.Enum):
 
1
  from __future__ import annotations
2
 
 
3
  import contextlib
4
  import enum
5
  import json
6
  import logging
7
+ from typing import Any, Dict, Generator, Iterable, List, Optional, Tuple, Type
 
 
8
 
 
9
  import pandas as pd
10
  import sqlalchemy
11
  from langchain.docstore.document import Document
12
  from langchain.schema.embeddings import Embeddings
13
  from langchain.utils import get_from_dict_or_env
14
  from langchain.vectorstores.base import VectorStore
 
15
  from pgvector.sqlalchemy import Vector
16
  from sqlalchemy import delete, text
17
+ from sqlalchemy.orm import Session, declarative_base
18
 
19
 
20
  class DistanceStrategy(str, enum.Enum):