Spaces:
Sleeping
Sleeping
Léo Bourrel
commited on
Commit
·
5ac52b0
1
Parent(s):
f0215ae
clean: isort
Browse files- app.py +1 -3
- 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
|
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 |
|