from langchain_azure_ai.chat_models import AzureAIChatCompletionsModel
from pydantic import BaseModel, Field
import os

llm = AzureAIChatCompletionsModel(
    endpoint=os.getenv("GITHUB_INFERENCE_ENDPOINT"),
    credential=os.getenv("GITHUB_TOKEN"),
    model="gpt-4o",
)

class ContactInfo(BaseModel):
    name: str = Field(description="The name of the person")
    email: str = Field(description="The email of the person")
    phone: str = Field(description="The phone number of the person")

agent = create_agent(
    model = gllm_4_1,
    tools = [search_tool],
    response_format=ProviderStrategy(ContactInfo)
)

agent.invoke({"messages": [HumanMessage(content="Extract contact info from: John Doe, john@example.com, (555) 123-4567")]})