fix tests

This commit is contained in:
Adil Hafeez 2025-03-25 21:02:54 -07:00
parent b73b177b3b
commit 49342c3c42
No known key found for this signature in database
GPG key ID: 9B18EF7691369645

View file

@ -220,7 +220,10 @@ impl HttpContext for StreamContext {
}));
} else {
self.select_llm_provider();
self.add_http_request_header(ARCH_ROUTING_HEADER, &self.llm_provider().name);
self.add_http_request_header(
ARCH_ROUTING_HEADER,
&self.llm_provider().provider_interface.to_string(),
);
if let Err(error) = self.modify_auth_headers() {
// ensure that the provider has an endpoint if the access key is missing else return a bad request
if self.llm_provider.as_ref().unwrap().endpoint.is_none() && !use_agent_orchestrator