MJannik commited on
Commit
24d947d
·
verified ·
1 Parent(s): bd46d25

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -8
app.py CHANGED
@@ -3,6 +3,11 @@ import json
3
  import uuid
4
  from langfuse import Langfuse
5
  import os
 
 
 
 
 
6
 
7
  os.environ["LANGFUSE_PUBLIC_KEY"] = os.getenv('LANGFUSE_PUBLIC_KEY')
8
  os.environ["LANGFUSE_SECRET_KEY"] = os.getenv('LANGFUSE_SECRET_KEY')
@@ -24,12 +29,6 @@ def set_new_session_id():
24
  # Initialize
25
  set_new_session_id()
26
 
27
- # Langfuse decorator
28
- from langfuse.decorators import observe, langfuse_context
29
- # Optional: automated instrumentation via OpenAI SDK integration
30
- # See note above regarding alternative implementations
31
- from langfuse.openai import openai
32
-
33
  # Global reference for the current trace_id which is used to later add user feedback
34
  current_trace_id = None
35
 
@@ -60,9 +59,9 @@ async def create_response(
60
  # Get completion via OpenAI SDK
61
  # Auto-instrumented by Langfuse via the import, see alternative in note above
62
  response = {"role": "assistant", "content": ""}
63
- oai_response = openai.chat.completions.create(
64
  messages=history,
65
- model="gpt-4o-mini",
66
  )
67
  response["content"] = oai_response.choices[0].message.content or ""
68
 
 
3
  import uuid
4
  from langfuse import Langfuse
5
  import os
6
+ # Langfuse decorator
7
+ from langfuse.decorators import observe, langfuse_context
8
+ # Optional: automated instrumentation via OpenAI SDK integration
9
+ # See note above regarding alternative implementations
10
+ from langfuse.openai import openai, OpenAI
11
 
12
  os.environ["LANGFUSE_PUBLIC_KEY"] = os.getenv('LANGFUSE_PUBLIC_KEY')
13
  os.environ["LANGFUSE_SECRET_KEY"] = os.getenv('LANGFUSE_SECRET_KEY')
 
29
  # Initialize
30
  set_new_session_id()
31
 
 
 
 
 
 
 
32
  # Global reference for the current trace_id which is used to later add user feedback
33
  current_trace_id = None
34
 
 
59
  # Get completion via OpenAI SDK
60
  # Auto-instrumented by Langfuse via the import, see alternative in note above
61
  response = {"role": "assistant", "content": ""}
62
+ oai_response = client.chat.completions.create(
63
  messages=history,
64
+ model="tgi",
65
  )
66
  response["content"] = oai_response.choices[0].message.content or ""
67