src/main/java/com/example/chat/logic/component/EquationWizard.java - develop - vorotov.a.vikt/MathEquGen

src/main/java/com/example/chat/logic/component/EquationWizard.java
62 строки3 KB
VorotovAlexey
разные доработки
10 дек 2025, 09:58
10 дек 2025, 09:58f42a8b8
package com.example.chat.logic.component;

import com.example.chat.logic.aiservice.EquationChecker;
import com.example.chat.logic.aiservice.EquationGenerator;
import com.example.chat.logic.aiservice.EquationAgent;
import com.example.chat.logic.service.FileGenerator;
import com.example.chat.logic.tool.EquationsGeneratorAndCorrectorTool;
import com.example.chat.logic.tool.EquationsSupplerTool;
import com.example.chat.logic.tool.ExamplesKeeperTool;
import dev.langchain4j.invocation.InvocationParameters;
import lombok.Getter;
import org.springframework.beans.factory.ObjectProvider;
import org.springframework.context.annotation.Scope;
import org.springframework.stereotype.Component;

@Component
@Scope("prototype")
public class EquationWizard {

    @Getter
    private final EquationsKeeper equationsKeeper;
    @Getter
    private final EquationAgent equationAgent;
    private final EquationsSupplerTool equationsSupplerTool;
    @Getter
    private final InvocationParameters invocationParameters = new InvocationParameters();
    @Getter
    private final Transceiver transceiver;
    private final ExamplesKeeper examplesKeeper;

    public EquationWizard(ObjectProvider<EquationsKeeper> equationsKeeperProvider,
                          EquationGenerator equationGenerator,
                          ObjectProvider<EquationsGeneratorAndCorrectorTool> equationsGeneratorAndSaverProvider,
                          ObjectProvider<EquationAgent> equationRobotProvider,
                          ObjectProvider<EquationsSupplerTool> equationsSupplerProvider,
                          ObjectProvider<Transceiver> transceiverProvider,
                          FileGenerator fileGenerator,
                          EquationChecker equationChecker,
                          ObjectProvider<ExamplesKeeper> examplesKeeperProvider,
                          ObjectProvider<ExamplesKeeperTool> examplesKeeperToolProvider) {
        this.equationsKeeper = equationsKeeperProvider.getObject();
        this.transceiver = transceiverProvider.getObject();
        var equationsGeneratorAndSaver = equationsGeneratorAndSaverProvider.getObject(this.equationsKeeper, equationGenerator, equationChecker);
        this.equationsSupplerTool = equationsSupplerProvider.getObject(this.equationsKeeper, this.transceiver, fileGenerator);
        this.examplesKeeper = examplesKeeperProvider.getObject();
        var examplesKeeperTool = examplesKeeperToolProvider.getObject(this.examplesKeeper);
        this.equationAgent = equationRobotProvider.getObject(equationsGeneratorAndSaver, this.equationsSupplerTool, examplesKeeperTool);
    }

    public EquationsSupplerTool getEquationsSuppler() {
        return equationsSupplerTool;
    }

    public void setExamples(String examples) {
        this.examplesKeeper.setExamples(examples);
    }

    public void addInvocationParameter(String key, Object value) {
        invocationParameters.put(key, value);
    }

}