LiteLLM Compatible Backend
# During initialization
if config.backend == "google":
config.backend = "gemini" # Convert backend name for compatibility
prefix = f"{config.backend}/"
if not config.name.startswith(prefix):
self.llms.append(prefix + config.name)completion_kwargs = {
"messages": messages,
"temperature": temperature,
"max_tokens": max_tokens
}
completed_response = completion(model=model, **completion_kwargs)
return completed_response.choices[0].message.content, TrueLast updated