tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-07-14 01:07:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

(test) langfuse logging with custom gen name

Browse Source
This commit is contained in:
ishaan-jaff
2023-10-16 21:34:44 -07:00
parent 5135b77c43
commit 730ee6a7ae
1 changed files with 53 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/tests/test_langfuse.py
+53 -1
View File
@@ -19,12 +19,64 @@ def test_langfuse_logging():
messages=[{
"role": "user",
"content": "Hi 👋 - i'm claude"
}])
}],
max_tokens=10,
)
print(response)
except Exception as e:
print(e)
test_langfuse_logging()
def test_langfuse_logging_custom_generation_name():
try:
response = completion(model="gpt-3.5-turbo",
messages=[{
"role": "user",
"content": "Hi 👋 - i'm claude"
}],
max_tokens=10,
metadata = {
"generation_name": "litellm-ishaan-gen"
}
)
print(response)
except Exception as e:
print(e)
test_langfuse_logging_custom_generation_name()
# def test_langfuse_logging_function_calling():
# function1 = [
# {
# "name": "get_current_weather",
# "description": "Get the current weather in a given location",
# "parameters": {
# "type": "object",
# "properties": {
# "location": {
# "type": "string",
# "description": "The city and state, e.g. San Francisco, CA",
# },
# "unit": {"type": "string", "enum": ["celsius", "fahrenheit"]},
# },
# "required": ["location"],
# },
# }
# ]
# try:
# response = completion(model="gpt-3.5-turbo",
# messages=[{
# "role": "user",
# "content": "what's the weather outside"
# }],
# functions=function1,
# )
# print(response)
# except Exception as e:
# print(e)
# test_langfuse_logging_function_calling()