server.py — langchain Source File

Source Code

"""A mock Robot server."""

from enum import Enum
from typing import Annotated, Any
from uuid import uuid4

import uvicorn
from fastapi import FastAPI, HTTPException, Query
from fastapi.middleware.cors import CORSMiddleware
from fastapi.openapi.utils import get_openapi
from pydantic import BaseModel, Field

PORT = 7289

app = FastAPI()
origins = [
    "http://localhost",
    "http://localhost:8000",
    "http://127.0.0.1",
    "http://127.0.0.1:8000",
]

app.add_middleware(
    CORSMiddleware,
    allow_origins=origins,
    allow_credentials=True,
    allow_methods=["*"],
    allow_headers=["*"],
)
PASS_PHRASE = str(uuid4())

_ROBOT_LOCATION = {"x": 0, "y": 0, "z": 0}


class StateItems(str, Enum):
    location = "location"
    walking = "walking"
    speed = "speed"
    direction = "direction"
    style = "style"
    cautiousness = "cautiousness"
    jumping = "jumping"
    destruct = "destruct"


_ROBOT_STATE = {
    "location": _ROBOT_LOCATION,
    "walking": False,
    "speed": 0,
    "direction": "north",
    "style": "normal",
    "cautiousness": "medium",
    "jumping": False,
    "destruct": False,
}


class Direction(str, Enum):
    north = "north"
    south = "south"
// ... (147 more lines)