diff --git a/litellm/tests/test_langfuse.py b/litellm/tests/test_langfuse.py
index ce23ab8e9f..fe5a35196f 100644
--- a/litellm/tests/test_langfuse.py
+++ b/litellm/tests/test_langfuse.py
@@ -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()
+