Skip to content

Commit

Permalink
add agents codes
Browse files Browse the repository at this point in the history
  • Loading branch information
ks6088ts committed Jun 9, 2024
1 parent 0ff308e commit cf381b1
Show file tree
Hide file tree
Showing 6 changed files with 69 additions and 0 deletions.
2 changes: 2 additions & 0 deletions backend/fastapi.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
from fastapi import FastAPI
from fastapi.openapi.utils import get_openapi

from backend.routers import agents as agents_router
from backend.routers import azure_ai_document_intelligence as azure_ai_document_intelligence_router
from backend.routers import azure_ai_speech as azure_ai_speech_router
from backend.routers import azure_ai_vision as azure_ai_vision_router
Expand All @@ -23,6 +24,7 @@
azure_storage_queue_router.router,
azure_cosmos_db_router.router,
azure_ai_speech_router.router,
agents_router.router,
]:
app.include_router(router)

Expand Down
10 changes: 10 additions & 0 deletions backend/internals/agents.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,10 @@
from logging import getLogger

from backend.settings.agents import Settings

logger = getLogger(__name__)


class Client:
def __init__(self, settings: Settings) -> None:
self.settings = settings
31 changes: 31 additions & 0 deletions backend/routers/agents.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,31 @@
from logging import getLogger

from fastapi import APIRouter, status
from fastapi.responses import JSONResponse

from backend.internals.agents import Client
from backend.schemas import agents as agents_schemas
from backend.settings.agents import Settings

logger = getLogger(__name__)

client = Client(
settings=Settings(),
)

router = APIRouter(
prefix="/agents",
tags=["agents"],
responses={404: {"description": "Not found"}},
)


@router.post(
"/hello",
status_code=200,
)
async def hello(request: agents_schemas.HelloRequest):
return JSONResponse(
status_code=status.HTTP_200_OK,
content=request.model_dump(),
)
9 changes: 9 additions & 0 deletions backend/schemas/agents.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
from logging import getLogger

from pydantic import BaseModel

logger = getLogger(__name__)


class HelloRequest(BaseModel):
prompt: str
13 changes: 13 additions & 0 deletions backend/settings/agents.py
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
from pydantic_settings import BaseSettings, SettingsConfigDict


class Settings(BaseSettings):
agents_azure_openai_endpoint: str = "https://<aoai-name>.openai.azure.com"
agents_azure_openai_api_key: str = "<aoai-api-key>"
agents_azure_openai_api_version: str = "2024-05-01-preview"
agents_azure_openai_gpt_model: str = "gpt-4o"

model_config = SettingsConfigDict(
env_file="settings/agents.env",
env_file_encoding="utf-8",
)
4 changes: 4 additions & 0 deletions settings/agents.env.sample
Original file line number Diff line number Diff line change
@@ -0,0 +1,4 @@
AGENTS_AZURE_OPENAI_ENDPOINT = "https://<aoai-name>.openai.azure.com"
AGENTS_AZURE_OPENAI_API_KEY = "<aoai-api-key>"
AGENTS_AZURE_OPENAI_API_VERSION = "2024-05-01-preview"
AGENTS_AZURE_OPENAI_GPT_MODEL = "gpt-4o"

0 comments on commit cf381b1

Please sign in to comment.