diff --git a/litellm/proxy/proxy_config.yaml b/litellm/proxy/proxy_config.yaml index 42e77475f7d..3f3b0858ef2 100644 --- a/litellm/proxy/proxy_config.yaml +++ b/litellm/proxy/proxy_config.yaml @@ -17,9 +17,7 @@ model_list: general_settings: master_key: sk-1234 - litellm_key_header_name: "X-Litellm-Key" litellm_settings: - cache: true - callbacks: ["otel"] + success_callback: ["langsmith"] diff --git a/litellm/tests/test_langsmith.py b/litellm/tests/test_langsmith.py index 603a8370da2..5f3837b264c 100644 --- a/litellm/tests/test_langsmith.py +++ b/litellm/tests/test_langsmith.py @@ -1,19 +1,20 @@ -import sys -import os import io +import os +import sys sys.path.insert(0, os.path.abspath("../..")) -from litellm import completion import litellm +from litellm import completion -litellm.success_callback = ["langsmith"] litellm.set_verbose = True import time def test_langsmith_logging(): try: + litellm.set_verbose = True + litellm.success_callback = ["langsmith"] response = completion( model="claude-instant-1.2", messages=[{"role": "user", "content": "what llm are u"}], @@ -21,6 +22,7 @@ def test_langsmith_logging(): temperature=0.2, ) print(response) + time.sleep(3) except Exception as e: print(e) @@ -30,17 +32,15 @@ def test_langsmith_logging(): def test_langsmith_logging_with_metadata(): try: + litellm.success_callback = ["langsmith"] response = completion( model="gpt-3.5-turbo", messages=[{"role": "user", "content": "what llm are u"}], max_tokens=10, temperature=0.2, - metadata={ - "run_name": "litellmRUN", - "project_name": "litellm-completion", - }, ) print(response) + time.sleep(3) except Exception as e: print(e) @@ -50,21 +50,19 @@ def test_langsmith_logging_with_metadata(): def test_langsmith_logging_with_streaming_and_metadata(): try: + litellm.success_callback = ["langsmith"] response = completion( model="gpt-3.5-turbo", messages=[{"role": "user", "content": "what llm are u"}], max_tokens=10, temperature=0.2, - metadata={ - "run_name": "litellmRUN", - "project_name": "litellm-completion", - }, stream=True, ) for chunk in response: continue + time.sleep(3) except Exception as e: print(e) -test_langsmith_logging_with_streaming_and_metadata() +# test_langsmith_logging_with_streaming_and_metadata()