from fastapi import FastAPI


class ModelName(str, Enum):
    alexnet = "alexnet"
    resnet = "resnet"
    lenet = "lenet"


app = FastAPI()


@app.get("/models/{model_name}")
async def get_model(model_name: ModelName):
    if model_name is ModelName.alexnet:
        return {"model_name": model_name, "message": "Deep Learning FTW!"}

    if model_name.value == "lenet":

    @Test
    public void test_setModel() {
        final LlmChatResponse response = new LlmChatResponse();

        response.setModel("llama3");
        assertEquals("llama3", response.getModel());

        response.setModel("gpt-3.5-turbo");
        assertEquals("gpt-3.5-turbo", response.getModel());
    }

    @Test
    public void test_nullValues() {

                raise ValueError("name must end in A")
            return name

    async def get_model_c() -> ModelC:
        return ModelC(username="test-user", password="test-password")

    @app.get("/model/{name}", response_model=ModelA)
    async def get_model_a(name: str, model_c=Depends(get_model_c)):
        return {
            "name": name,
            "description": "model-a-desc",
            "foo": model_c,

    }

    @Test
    public void test_setModel() {
        final LlmChatRequest request = new LlmChatRequest();

        final LlmChatRequest result = request.setModel("gpt-4");

        assertSame(request, result);
        assertEquals("gpt-4", request.getModel());
    }

    @Test
    public void test_setMaxTokens() {
        final LlmChatRequest request = new LlmChatRequest();

    }

    /**
     * Gets the model name.
     *
     * @return the model name
     */
    public String getModel() {
        return model;
    }

    /**
     * Sets the model name.
     *
     * @param model the model name
     */
    public void setModel(final String model) {
        this.model = model;
    }

     * Gets the model name.
     *
     * @return the model name
     */
    public String getModel() {
        return model;
    }

    /**
     * Sets the model name.
     *
     * @param model the model name
     * @return this request for method chaining
     */
    public LlmChatRequest setModel(final String model) {
        this.model = model;
        return this;
    }

    /**

        assertNull(result);
    }

    @Test
    public void test_evaluate_withValidModelName() {
        // Test with valid model name
        final String testModelName = "test_model";
        final int testWindowSize = 100;

        ComponentUtil.setFessConfig(new FessConfig.SimpleImpl() {
            @Override
            public String getLtrModelName() {

            deferDeleteFile(consumer);

            project.addAttachedArtifact(createConsumerPomArtifact(project, consumer, session));
        } else if (project.getModel().getDelegate().isRoot()) {
            throw new IllegalStateException(
                    "The use of the root attribute on the model requires the buildconsumer feature to be active");
        }
    }

    }

    @Override
    public Model build(RepositorySystemSession session, MavenProject project, ModelSource src)
            throws ModelBuilderException {
        Model model = project.getModel().getDelegate();
        boolean flattenEnabled = Features.consumerPomFlatten(session.getConfigProperties());
        String packaging = model.getPackaging();

            logger.debug("Initialized {} with timeout: {}ms", getClass().getSimpleName(), timeout);
        }
        logger.info("[LLM] {} initialized. model={}, timeout={}ms, maxConcurrent={}", getName(), getModel(), getTimeout(),
                getMaxConcurrentRequests());

        concurrencyLimiter = new Semaphore(getMaxConcurrentRequests());

        startAvailabilityCheck();
    }

    /**