Add name validation for OAI error prevention (#310)

* Name validation * Fix build * Add validation only for oai provider * Use preexisting * Fix formatting --------- Co-authored-by: Zubeir Mohamed <[email protected]>
microsoft · Aug 1, 2024 · a238a0b · a238a0b
1 parent 7367b0a
commit a238a0b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/python/src/agnext/components/models/_openai_client.py b/python/src/agnext/components/models/_openai_client.py
@@ -120,6 +120,7 @@ def type_to_role(message: LLMMessage) -> ChatCompletionRole:
 
 
 def user_message_to_oai(message: UserMessage) -> ChatCompletionUserMessageParam:
+    assert_valid_name(message.source)
     if isinstance(message.content, str):
         return ChatCompletionUserMessageParam(
             content=message.content,
@@ -177,6 +178,7 @@ def tool_message_to_oai(
 def assistant_message_to_oai(
     message: AssistantMessage,
 ) -> ChatCompletionAssistantMessageParam:
+    assert_valid_name(message.source)
     if isinstance(message.content, list):
         return ChatCompletionAssistantMessageParam(
             tool_calls=[func_call_to_oai(x) for x in message.content],