set up commands for using kiota to generate client codes

Author: ks6088ts

.pre-commit-config.yaml

@@ -13,3 +13,4 @@ repos:
     rev: 24.2.0
     hooks:
     -   id: black
+exclude: 'client/.*'

Makefile

@@ -30,9 +30,15 @@ format: ## format code
 	poetry run isort .
 	poetry run black . --verbose
 
+.PHONY: fix
+fix: format ## apply auto-fixes
+	poetry run ruff check --fix
+
 .PHONY: lint
 lint: ## lint
-	poetry run ruff check .
+	poetry run ruff check . \
+		--exclude client \
+		--respect-gitignore
 	shellcheck scripts/*.sh
 
 .PHONY: test
@@ -121,3 +127,14 @@ azure-functions-functionapp-deploy: ## deploy Azure Functions App
 .PHONY: generate-openapi-spec
 generate-openapi-spec: ## generate OpenAPI spec
 	poetry run python main.py generate-openapi-spec
+
+.PHONY: generate-openapi-client
+generate-openapi-client: ## generate OpenAPI client
+	@kiota generate \
+		--language Python \
+		--class-name ApiClient \
+		--namespace-name client \
+		--openapi ./specs/openapi.json \
+		--output ./client
+	@echo "Get the list of dependencies"
+	@kiota info -d ./specs/openapi.json --language Python

backend/fastapi.py

@@ -1,4 +1,5 @@
 from fastapi import FastAPI
+from fastapi.openapi.utils import get_openapi
 
 from backend.routers import azure_openai as azure_openai_router
 from backend.routers import document_intelligence as document_intelligence_router
@@ -9,3 +10,29 @@
 
 app.include_router(azure_openai_router.router)
 app.include_router(document_intelligence_router.router)
+
+
+def custom_openapi():
+    if app.openapi_schema:
+        return app.openapi_schema
+    openapi_schema = get_openapi(
+        title="Azure AI Services Solutions",
+        version="0.0.1",
+        description="This contains a collection of solutions that leverage Azure AI services.",
+        routes=app.routes,
+        servers=[
+            {
+                "url": "http://localhost:8000",
+            }
+        ],
+    )
+    openapi_schema["info"]["x-logo"] = {
+        "url": "https://news.microsoft.com/wp-content/uploads/prod/2022/05/Microsoft-logo_rgb_c-gray-1024x459.png"
+    }
+    openapi_schema["openapi"] = "3.0.0"
+    app.openapi_schema = openapi_schema
+    return app.openapi_schema
+
+
+app.openapi_version = "3.0.0"  # to use Kiota, downgrade to 3.0.0 (ref. https://github.com/microsoft/kiota/issues/3914)
+app.openapi = custom_openapi

docs/README.md

@@ -3,6 +3,7 @@
 ## Common
 
 - [Typer](https://typer.tiangolo.com/#installation)
+- [Kiota > Generate tailored Python and PHP API clients for any API with Kiota](https://devblogs.microsoft.com/microsoft365dev/generate-tailored-python-and-php-api-clients-for-any-api-with-kiota/)
 
 ## Backend
 

frontend/solutions/sandbox.py

@@ -4,11 +4,37 @@
 
 import streamlit as st
 
-from frontend.solutions.utilities import http_get, http_post
+from frontend.solutions.utilities import http_get
 
 logger = logging.getLogger(__name__)
 
 
+async def chat_completions_post(
+    backend_url: str,
+    prompt: str,
+):
+    from kiota_abstractions.authentication.anonymous_authentication_provider import AnonymousAuthenticationProvider
+    from kiota_http.httpx_request_adapter import HttpxRequestAdapter
+
+    from client.api_client import ApiClient
+    from client.models.chat_completion_request import ChatCompletionRequest
+
+    auth_provider = AnonymousAuthenticationProvider()
+    request_adapter = HttpxRequestAdapter(
+        authentication_provider=auth_provider,
+        base_url=backend_url,
+    )
+    client = ApiClient(request_adapter)
+
+    response = await client.azure_openai.chat_completions.post(
+        ChatCompletionRequest(
+            content=prompt,
+            stream=False,
+        ),
+    )
+    return response.content
+
+
 def start(
     backend_url: str,
     log_level: int,
@@ -42,12 +68,9 @@ def start(
         try:
             with st.spinner("Calling API..."):
                 response = asyncio.run(
-                    http_post(
-                        url=urljoin(base=backend_url, url="/azure_openai/chat_completions/"),
-                        data={
-                            "content": prompt,
-                            "stream": False,
-                        },
+                    chat_completions_post(
+                        backend_url=backend_url,
+                        prompt=prompt,
                     )
                 )
             st.write(response)