Spaces:
Sleeping
Sleeping
Update pipeline.py
Browse files- pipeline.py +5 -0
pipeline.py
CHANGED
@@ -25,6 +25,11 @@ from refusal_chain import get_refusal_chain
|
|
25 |
from tailor_chain import get_tailor_chain
|
26 |
from prompts import classification_prompt, refusal_prompt, tailor_prompt
|
27 |
|
|
|
|
|
|
|
|
|
|
|
28 |
# Initialize Mistral API client
|
29 |
mistral_api_key = os.environ.get("MISTRAL_API_KEY")
|
30 |
client = Mistral(api_key=mistral_api_key)
|
|
|
25 |
from tailor_chain import get_tailor_chain
|
26 |
from prompts import classification_prompt, refusal_prompt, tailor_prompt
|
27 |
|
28 |
+
|
29 |
+
LANGSMITH_TRACING=true
|
30 |
+
LANGSMITH_ENDPOINT="https://api.smith.langchain.com"
|
31 |
+
LANGSMITH_API_KEY=os.environ.get("LANGSMITH_API_KEY")
|
32 |
+
LANGSMITH_PROJECT=os.environ.get("LANGCHAIN_PROJECT")
|
33 |
# Initialize Mistral API client
|
34 |
mistral_api_key = os.environ.get("MISTRAL_API_KEY")
|
35 |
client = Mistral(api_key=mistral_api_key)
|