semantic_kernel_quickstart_openai_prompt_invocation.py

1import asyncio
2
3from semantic_kernel import Kernel
4from semantic_kernel.connectors.ai.open_ai import AzureChatCompletion, OpenAIChatCompletion
5from semantic_kernel.functions import KernelArguments
6
7async def main():
8    # Initialize the kernel
9    kernel = Kernel()
10
11    # Add a chat completion service (Choose either AzureOpenAI or OpenAI)
12    # For Azure OpenAI:
13    # kernel.add_service(
14    #     AzureChatCompletion(
15    #         deployment_name="your_deployment_name",
16    #         api_key="your_api_key",
17    #         endpoint="your_endpoint"
18    #     )
19    # )
20    
21    # For OpenAI:
22    kernel.add_service(
23        OpenAIChatCompletion(
24            ai_model_id="gpt-3.5-turbo",
25            api_key="your_api_key"
26        )
27    )
28
29    # Define a prompt and create a function
30    prompt = "What is the capital of France?"
31    
32    # Execute the prompt
33    result = await kernel.invoke_prompt(prompt)
34    
35    print(result)
36
37# Run the main function
38if __name__ == "__main__":
39    asyncio.run(main())