respecting OPENAI_MODEL_NAME

This commit is contained in:
João Moura
2024-09-22 11:20:54 -03:00
parent c055c35361
commit c4167a5517

View File

@@ -119,7 +119,8 @@ class Agent(BaseAgent):
def post_init_setup(self):
self.agent_ops_agent_name = self.role
self.llm = (
getattr(self.llm, "model_name", None)
os.environ.get("OPENAI_MODEL_NAME")
or getattr(self.llm, "model_name", None)
or getattr(self.llm, "deployment_name", None)
or self.llm
)