phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-27 03:34:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

(fix) llmonitor track function calls

Browse source
This commit is contained in:
ishaan-jaff 2023-10-20 09:05:08 -07:00
parent 123c0f41f8
commit 8e9340f3b9
2 changed files with 40 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

38
litellm/tests/test_llmonitor_integration.py
View file

@ -38,5 +38,39 @@ def test_embedding_openai():
print(e)
test_chat_openai()
test_embedding_openai()
# test_chat_openai()
# test_embedding_openai()
def test_llmonitor_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 in boston"
}],
temperature=0.1,
functions=function1,
)
print(response)
except Exception as e:
print(e)
# test_llmonitor_logging_function_calling()