Update app.py

app.py

@@ -77,24 +77,18 @@ from langchain.memory import ConversationSummaryBufferMemory,ConversationBufferM
 from langchain.schema import Document
 
 
-import os
-import base64
-LANGFUSE_PUBLIC_KEY= os.environ.get("LANGFUSE_PUBLIC_KEY")
-LANGFUSE_SECRET_KEY= os.environ.get("LANGFUSE_SECRET_KEY")
-LANGFUSE_AUTH=base64.b64encode(f"{LANGFUSE_PUBLIC_KEY}:{LANGFUSE_SECRET_KEY}".encode()).decode()
-#os.environ["OTEL_EXPORTER_OTLP_ENDPOINT"] = "https://cloud.langfuse.com/api/public/otel" # EU data region
-os.environ["OTEL_EXPORTER_OTLP_ENDPOINT"] = "https://us.cloud.langfuse.com/api/public/otel" # US data region
-os.environ["OTEL_EXPORTER_OTLP_HEADERS"] = f"Authorization=Basic {LANGFUSE_AUTH}"
+from langfuse import Langfuse
 
+langfuse = Langfuse(
+  secret_key=os.environ.get("LANGFUSE_SECRET_KEY"),
+  public_key=os.environ.get("LANGFUSE_PUBLIC_KEY"),
+  host="https://us.cloud.langfuse.com"
+)
+from langfuse.decorators import observe
 
-from opentelemetry.sdk.trace import TracerProvider
-from openinference.instrumentation.smolagents import SmolagentsInstrumentor
-from opentelemetry.exporter.otlp.proto.http.trace_exporter import OTLPSpanExporter
-from opentelemetry.sdk.trace.export import SimpleSpanProcessor 
-trace_provider = TracerProvider()
-trace_provider.add_span_processor(SimpleSpanProcessor(OTLPSpanExporter())) 
-SmolagentsInstrumentor().instrument(tracer_provider=trace_provider)
-
+LANGFUSE_SECRET_KEY=os.environ.get("LANGFUSE_SECRET_KEY")
+LANGFUSE_PUBLIC_KEY=os.environ.get("LANGFUSE_PUBLIC_KEY")
+LANGFUSE_HOST="https://us.cloud.langfuse.com"
 
 
 
@@ -620,7 +614,7 @@ chain = ConversationalRetrievalChain.from_llm(
 
 
 # This below is for the interface 
-
+@observe()
 def submit_message(prompt, prompt_template, temperature, max_tokens, context_length, state):