Minor misc ux improvements (#3966)

* minor misc ux

* nit

* k

* quick nit

* k
This commit is contained in:
pablonyx
2025-02-12 20:43:11 -08:00
committed by GitHub
parent 2850048c6b
commit 9db7b67a6c
24 changed files with 189 additions and 60 deletions

View File

@@ -396,9 +396,14 @@ class DefaultMultiLLM(LLM):
self._record_call(processed_prompt)
try:
print(
"model is",
f"{self.config.model_provider}/{self.config.deployment_name or self.config.model_name}",
)
return litellm.completion(
mock_response=MOCK_LLM_RESPONSE,
# model choice
# model="openai/gpt-4",
model=f"{self.config.model_provider}/{self.config.deployment_name or self.config.model_name}",
# NOTE: have to pass in None instead of empty string for these
# otherwise litellm can have some issues with bedrock