Yury Z.

from openinference.instrumentation.openai import OpenAIInstrumentor

from phoenix.otel import register

tracer_provider = register()
OpenAIInstrumentor().instrument(tracer_provider=tracer_provider)

def main(input) -> str:
    chain = create_chain()
    session_id_requests = f"TestBackend{random.randint(0, 1000000000)}"
    conversation_id_requests = f"TestBackendConversation{random.randint(0, 1000000000)}"
    upm_id_requests = f"TestBackendUpmId{random.randint(0, 1000000000)}"

    user_input = input["question"]
      
    data_input = {
                "language": "en",
                "marketplace": "US",
                "channel": "test_channel",
                "session_id": session_id_requests,
                "conversation_id": conversation_id_requests,
                "generated_text_response": False,
                "text": user_input,
                "upm_id": upm_id_requests,
                  }
    chain_out = chain.invoke(Request(**data_input))

    return chain_out.answer