Spaces:
Runtime error
Runtime error
File size: 3,198 Bytes
b044427 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 |
import fastapi
from typing import Optional
import copy
from starlette.middleware.sessions import SessionMiddleware
from fastapi import Request
class Tool(fastapi.FastAPI):
""" Tool is inherited from FastAPI class, thus:
1. It can act as a server
2. It has get method, you can use Tool.get method to bind a function to an url
3. It can be easily mounted to another server
4. It has a list of sub-routes, each route is a function
Diagram:
Root API server (ToolServer object)
│
├───── "./weather": Tool object
│ ├── "./get_weather_today": function_get_weather(location: str) -> str
│ ├── "./get_weather_forecast": function_get_weather_forcast(location: str, day_offset: int) -> str
│ └── "...more routes"
├───── "./wikidata": Tool object
│ ├── "... more routes"
└───── "... more routes"
"""
def __init__(
self,
tool_name : str,
description : str,
name_for_human : Optional[str] = None,
name_for_model : Optional[str] = None,
description_for_human : Optional[str] = None,
description_for_model : Optional[str] = None,
logo_url : Optional[str] = None,
author_github : Optional[str] = None,
contact_email : str = "",
legal_info_url : str = "",
version : str = "0.1.0",
):
super().__init__(
title=tool_name,
description=description,
version=version,
)
if name_for_human is None:
name_for_human = tool_name
if name_for_model is None:
name_for_model = name_for_human
if description_for_human is None:
description_for_human = description
if description_for_model is None:
description_for_model = description_for_human
self.api_info = {
"schema_version": "v1",
"name_for_human": name_for_human,
"name_for_model": name_for_model,
"description_for_human": description_for_human,
"description_for_model": description_for_model,
"auth": {
"type": "none",
},
"api": {
"type": "openapi",
"url": "/openapi.json",
"is_user_authenticated": False,
},
"author_github": author_github,
"logo_url": logo_url,
"contact_email": contact_email,
"legal_info_url": legal_info_url,
}
@self.get("/.well-known/ai-plugin.json", include_in_schema=False)
def get_api_info(request : fastapi.Request):
openapi_path = str(request.url).replace("/.well-known/ai-plugin.json", "/openapi.json")
info = copy.deepcopy(self.api_info)
info["api"]["url"] = str(openapi_path)
return info
self.add_middleware(
SessionMiddleware,
secret_key=tool_name,
session_cookie="session_{}".format(tool_name.replace(" ", "_")),
)
|