File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 1212# from email.policy import default as email_policy
1313import re
1414
15- chosen_model = os .environ [ "CHOSEN_MODEL" ]
15+ LLM_MODEL = os .getenv ( "LLM_MODEL" , "llava:7b" )
1616if os .environ .get ("MANUAL_TRACING" ):
1717 span_processor_endpoint = os .environ .get ("OTEL_ENDPOINT" )
1818 if span_processor_endpoint is None :
3434 LangchainInstrumentor ().instrument ()
3535
3636 # Service name is required for most backends
37- resource = Resource (
38- attributes = {SERVICE_NAME : f"LLM inference-{ chosen_model .lower ()} " }
39- )
37+ resource = Resource (attributes = {SERVICE_NAME : f"LLM inference-{ LLM_MODEL .lower ()} " })
4038
4139 trace_provider = TracerProvider (resource = resource )
4240 processor = BatchSpanProcessor (OTLPSpanExporter (endpoint = span_processor_endpoint ))
5452ollama_host = os .getenv ("OLLAMA_HOST" , "http://127.0.0.1:11434" )
5553if not ollama_host .startswith ("http://" ) and not ollama_host .startswith ("https://" ):
5654 ollama_host = "http://" + ollama_host
57- LLM_MODEL = os .getenv ("LLM_MODEL" , "llava:7b" )
55+ # LLM_MODEL = os.getenv("LLM_MODEL", "llava:7b")
5856
5957# create a single AsyncClient for reuse
6058client = AsyncClient (host = ollama_host )
You can’t perform that action at this time.
0 commit comments