Hi I'm trying to run UDF funtion and integarting it with azure open ai.
I'm getting the below error during invocation can someone please help?
{
"functionName": "AISuggestion",
"invocationId": "00000000-0000-0000-0000-000000000000",
"status": "Failed",
"errors": [
{
"errorCode": "WorkloadException",
"subErrorCode": "NotFound",
"message": "User data function: 'AISuggestion' invocation failed."
}
]
}
My code
------------------------------------------------------------------------------------------
import fabric.functions as fn
import logging
import openai
udf = fn.UserDataFunctions()
@udf.connection(argName="sqlDB", alias="MyFabricSQLDB")
@udf.function()
def AISuggestion(sqlDB: fn.FabricSqlConnection, company: str) -> str:
try:
logging.info('Triggering AISuggestion function.')
# Connect to SQL
try:
connection = sqlDB.connect()
cursor = connection.cursor()
except Exception as sql_conn_err:
return f"SQL connection failed: {str(sql_conn_err)}"
# Read company info
try:
cursor.execute("SELECT * FROM [dbo].[CompanyStatus] WHERE Company = ?", company)
record = cursor.fetchone()
if not record:
return f"No record found for company: {company}"
customer_name = record[0]
last_comment = record[10]
except Exception as query_err:
return f"Error fetching company data: {str(query_err)}"
# OpenAI Prompt
prompt = (
"Respond with a short plan that is under 240 characters: I work at Contoso Outdoors, "
"and we collaborate with influencers by offering them offers for custom designed bikes. "
f"Pretend we want to collab with the following influencer: {customer_name} from company: {company}. "
f"Here's a comment about their latest feedback: {last_comment}."
)
# Call Azure OpenAI
try:
deployment = "gpt-35-turbo" # or gpt-35-turbo
openai_client = openai.AzureOpenAI(
api_key='xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx',
api_version="2025-01-01-preview",
)
response = openai_client.chat.completions.create(
model=deployment,
messages=[
{"role": "user", "content": prompt}
]
)
result = response.choices[0].message.content
except Exception as ai_err:
return f"OpenAI call failed: {str(ai_err)}"
# Write result to SQL
try:
cursor.execute("SELECT * FROM [dbo].[AISuggestions] WHERE Company = ?", company)
if cursor.fetchone():
update_query = "UPDATE [dbo].[AISuggestions] SET [AI_suggestion] = ? WHERE [Company] = ?;"
cursor.execute(update_query, (result, company))
else:
insert_query = "INSERT INTO [dbo].[AISuggestions] (Name, Company, AI_suggestion) VALUES (?, ?, ?);"
cursor.execute(insert_query, (customer_name, company, result))
connection.commit()
except Exception as sql_write_err:
return f"SQL insert/update failed: {str(sql_write_err)}"
finally:
cursor.close()
connection.close()
return f"Successfully generated suggestion for {customer_name} from {company}."
except Exception as general_err:
return f"Unhandled error: {str(general_err)}"
