Léo Bourrel commited on
Commit
5ac52b0
·
1 Parent(s): f0215ae

clean: isort

Browse files
Files changed (2) hide show
  1. app.py +1 -3
  2. custom_pgvector.py +2 -1
app.py CHANGED
@@ -2,15 +2,13 @@ import json
2
  import os
3
 
4
  import sqlalchemy
5
- import sqlite_vss
6
  import streamlit as st
7
  import streamlit.components.v1 as components
8
- from langchain.llms import OpenAI
9
  from langchain.callbacks import get_openai_callback
10
  from langchain.chains import ConversationalRetrievalChain
11
  from langchain.chains.conversation.memory import ConversationBufferMemory
12
  from langchain.embeddings import GPT4AllEmbeddings
13
- from sqlalchemy import event
14
 
15
  from chat_history import insert_chat_history, insert_chat_history_articles
16
  from css import load_css
 
2
  import os
3
 
4
  import sqlalchemy
 
5
  import streamlit as st
6
  import streamlit.components.v1 as components
 
7
  from langchain.callbacks import get_openai_callback
8
  from langchain.chains import ConversationalRetrievalChain
9
  from langchain.chains.conversation.memory import ConversationBufferMemory
10
  from langchain.embeddings import GPT4AllEmbeddings
11
+ from langchain.llms import OpenAI
12
 
13
  from chat_history import insert_chat_history, insert_chat_history_articles
14
  from css import load_css
custom_pgvector.py CHANGED
@@ -4,7 +4,6 @@ import contextlib
4
  import enum
5
  import json
6
  import logging
7
- import struct
8
  from typing import Any, Dict, Generator, Iterable, List, Optional, Tuple, Type
9
 
10
  import pandas as pd
@@ -16,8 +15,10 @@ from langchain.vectorstores.base import VectorStore
16
  from pgvector.sqlalchemy import Vector
17
  from sqlalchemy import delete, text
18
  from sqlalchemy.orm import Session, declarative_base
 
19
  from utils import str_to_list
20
 
 
21
  class DistanceStrategy(str, enum.Enum):
22
  """Enumerator of the Distance strategies."""
23
 
 
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
 
15
  from pgvector.sqlalchemy import Vector
16
  from sqlalchemy import delete, text
17
  from sqlalchemy.orm import Session, declarative_base
18
+
19
  from utils import str_to_list
20
 
21
+
22
  class DistanceStrategy(str, enum.Enum):
23
  """Enumerator of the Distance strategies."""
24