Genie Agent integration issue

yj940525
New Contributor III

Hi, anyone from development team for Genie Agent integration?  i had an issue of using sample code of Genie Agent integration. The issue is that underlying code (databricks_ai_bridge/genie.py) cannot connect to url openaipublic.blob.core.windows.net for openai tokenization and calculate total count of query result string. Since our Databricks workspace sits on enclosed AWS VPC which doesn't have access to public internet (i guess there are other databricks users like us).  Just wondering whether this implementation can be done in another way without make API call to public internet.