Spaces:
Sleeping
Sleeping
corrected the import in the chain folder
Browse files- chain/classification_chain.py +1 -1
- chain/cleaner_chain.py +1 -1
- chain/refusal_chain.py +1 -1
- chain/tailor_chain.py +1 -1
- chain/tailor_chain_wellnessBrand.py +1 -1
chain/classification_chain.py
CHANGED
@@ -4,7 +4,7 @@ from langchain.chains import LLMChain
|
|
4 |
from langchain_groq import ChatGroq
|
5 |
|
6 |
# We'll import the classification_prompt from prompts.py
|
7 |
-
from
|
8 |
|
9 |
def get_classification_chain() -> LLMChain:
|
10 |
"""
|
|
|
4 |
from langchain_groq import ChatGroq
|
5 |
|
6 |
# We'll import the classification_prompt from prompts.py
|
7 |
+
from chain.prompts import classification_prompt
|
8 |
|
9 |
def get_classification_chain() -> LLMChain:
|
10 |
"""
|
chain/cleaner_chain.py
CHANGED
@@ -2,7 +2,7 @@
|
|
2 |
import os
|
3 |
from langchain.chains import LLMChain
|
4 |
from langchain_groq import ChatGroq
|
5 |
-
from
|
6 |
|
7 |
class CleanerChain(LLMChain):
|
8 |
def merge(self, kb: str, web: str) -> str:
|
|
|
2 |
import os
|
3 |
from langchain.chains import LLMChain
|
4 |
from langchain_groq import ChatGroq
|
5 |
+
from chain.prompts import cleaner_prompt
|
6 |
|
7 |
class CleanerChain(LLMChain):
|
8 |
def merge(self, kb: str, web: str) -> str:
|
chain/refusal_chain.py
CHANGED
@@ -3,7 +3,7 @@ import os
|
|
3 |
from langchain.chains import LLMChain
|
4 |
from langchain_groq import ChatGroq
|
5 |
|
6 |
-
from
|
7 |
|
8 |
def get_refusal_chain() -> LLMChain:
|
9 |
"""
|
|
|
3 |
from langchain.chains import LLMChain
|
4 |
from langchain_groq import ChatGroq
|
5 |
|
6 |
+
from chain.prompts import refusal_prompt
|
7 |
|
8 |
def get_refusal_chain() -> LLMChain:
|
9 |
"""
|
chain/tailor_chain.py
CHANGED
@@ -3,7 +3,7 @@ import os
|
|
3 |
from langchain.chains import LLMChain
|
4 |
from langchain_groq import ChatGroq
|
5 |
|
6 |
-
from
|
7 |
|
8 |
def get_tailor_chain() -> LLMChain:
|
9 |
"""
|
|
|
3 |
from langchain.chains import LLMChain
|
4 |
from langchain_groq import ChatGroq
|
5 |
|
6 |
+
from chain.prompts import tailor_prompt
|
7 |
|
8 |
def get_tailor_chain() -> LLMChain:
|
9 |
"""
|
chain/tailor_chain_wellnessBrand.py
CHANGED
@@ -3,7 +3,7 @@ import os
|
|
3 |
from langchain.chains import LLMChain
|
4 |
from langchain_groq import ChatGroq
|
5 |
|
6 |
-
from
|
7 |
|
8 |
def get_tailor_chain_wellnessBrand() -> LLMChain:
|
9 |
"""
|
|
|
3 |
from langchain.chains import LLMChain
|
4 |
from langchain_groq import ChatGroq
|
5 |
|
6 |
+
from chain.prompts import tailort_promptWellnessBrand1
|
7 |
|
8 |
def get_tailor_chain_wellnessBrand() -> LLMChain:
|
9 |
"""
|