datacipen commited on
Commit
a4d2937
·
verified ·
1 Parent(s): 41a4f7a

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -1
main.py CHANGED
@@ -6,7 +6,8 @@ import chainlit as cl
6
  from azure.ai.inference import ChatCompletionsClient
7
  from azure.ai.inference.models import SystemMessage, UserMessage
8
  from azure.core.credentials import AzureKeyCredential
9
- import pydantic
 
10
  os.environ["GITHUB_TOKEN"] = os.environ["GITHUB_TOKEN"]
11
 
12
  @cl.step(type="tool", show_input=True)
 
6
  from azure.ai.inference import ChatCompletionsClient
7
  from azure.ai.inference.models import SystemMessage, UserMessage
8
  from azure.core.credentials import AzureKeyCredential
9
+ from pydantic.dataclasses import rebuild_dataclass
10
+ pydantic.dataclasses.rebuild_dataclass(CodeSettings)
11
  os.environ["GITHUB_TOKEN"] = os.environ["GITHUB_TOKEN"]
12
 
13
  @cl.step(type="tool", show_input=True)