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)
|