Phoenix21 commited on
Commit
9583bff
·
verified ·
1 Parent(s): 55d7984

Update pipeline.py

Browse files
Files changed (1) hide show
  1. 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)