Spaces:
Sleeping
Sleeping
Commit
·
5ffa07c
1
Parent(s):
47a81d9
clean: isort
Browse files- sorbobotapp/chain.py +1 -2
- sorbobotapp/keyword_extraction.py +2 -2
- sorbobotapp/vector_store.py +2 -3
sorbobotapp/chain.py
CHANGED
@@ -1,11 +1,10 @@
|
|
1 |
import os
|
2 |
|
3 |
import sqlalchemy
|
|
|
4 |
from langchain.chains.conversation.memory import ConversationBufferMemory
|
5 |
from langchain.embeddings import GPT4AllEmbeddings
|
6 |
from langchain.llms import OpenAI
|
7 |
-
|
8 |
-
from conversation_retrieval_chain import CustomConversationalRetrievalChain
|
9 |
from vector_store import CustomVectorStore
|
10 |
|
11 |
|
|
|
1 |
import os
|
2 |
|
3 |
import sqlalchemy
|
4 |
+
from conversation_retrieval_chain import CustomConversationalRetrievalChain
|
5 |
from langchain.chains.conversation.memory import ConversationBufferMemory
|
6 |
from langchain.embeddings import GPT4AllEmbeddings
|
7 |
from langchain.llms import OpenAI
|
|
|
|
|
8 |
from vector_store import CustomVectorStore
|
9 |
|
10 |
|
sorbobotapp/keyword_extraction.py
CHANGED
@@ -1,9 +1,9 @@
|
|
1 |
from typing import Any
|
2 |
-
|
3 |
from langchain.chat_models import ChatOpenAI
|
4 |
from langchain.output_parsers import NumberedListOutputParser
|
5 |
from langchain.prompts import ChatPromptTemplate
|
6 |
-
|
7 |
|
8 |
query_template = """
|
9 |
You are a bi-lingual (french and english) linguistic teacher working at a top-tier university.
|
|
|
1 |
from typing import Any
|
2 |
+
|
3 |
from langchain.chat_models import ChatOpenAI
|
4 |
from langchain.output_parsers import NumberedListOutputParser
|
5 |
from langchain.prompts import ChatPromptTemplate
|
6 |
+
from utils import str_to_list
|
7 |
|
8 |
query_template = """
|
9 |
You are a bi-lingual (french and english) linguistic teacher working at a top-tier university.
|
sorbobotapp/vector_store.py
CHANGED
@@ -10,11 +10,10 @@ import sqlalchemy
|
|
10 |
from langchain.docstore.document import Document
|
11 |
from langchain.schema.embeddings import Embeddings
|
12 |
from langchain.vectorstores.base import VectorStore
|
13 |
-
from sqlalchemy import delete, text
|
14 |
-
from sqlalchemy.orm import Session
|
15 |
-
|
16 |
from models.article import Article
|
17 |
from models.distance import DistanceStrategy, distance_strategy_limit
|
|
|
|
|
18 |
from utils import str_to_list
|
19 |
|
20 |
DEFAULT_DISTANCE_STRATEGY = DistanceStrategy.COSINE
|
|
|
10 |
from langchain.docstore.document import Document
|
11 |
from langchain.schema.embeddings import Embeddings
|
12 |
from langchain.vectorstores.base import VectorStore
|
|
|
|
|
|
|
13 |
from models.article import Article
|
14 |
from models.distance import DistanceStrategy, distance_strategy_limit
|
15 |
+
from sqlalchemy import delete, text
|
16 |
+
from sqlalchemy.orm import Session
|
17 |
from utils import str_to_list
|
18 |
|
19 |
DEFAULT_DISTANCE_STRATEGY = DistanceStrategy.COSINE
|