quick fix for ollama prompt

Author: alizenhom

src/langtrace_python_sdk/instrumentation/ollama/patch.py

@@ -22,7 +22,10 @@ def traced_method(wrapped, instance, args, kwargs):
         service_provider = SERVICE_PROVIDERS["OLLAMA"]
         span_attributes = {
             **get_langtrace_attributes(version, service_provider),
-            **get_llm_request_attributes(kwargs, prompts=kwargs.get("messages", [])),
+            **get_llm_request_attributes(
+                kwargs,
+                prompts=kwargs.get("messages", None),
+            ),
             **get_llm_url(instance),
             SpanAttributes.LLM_PATH: api["ENDPOINT"],
             SpanAttributes.LLM_RESPONSE_FORMAT: kwargs.get("format"),

src/langtrace_python_sdk/utils/llm.py

@@ -100,7 +100,6 @@ def get_llm_request_attributes(kwargs, prompts=None, model=None):
             if "prompt" in kwargs
             else None
         )
-    print("PRMPT", prompts)
     top_k = (
         kwargs.get("n", None)
         or kwargs.get("k", None)