Merge branch 'eesast:dev' into dev

Author: jxd136

CAPI/cpp/API/include/API.h

@@ -75,6 +75,7 @@ class ILogic
     virtual bool BuildCharacter(THUAI9::CharacterType CharacterType, int32_t playerID) = 0;
     virtual bool ProduceGoods(THUAI9::GoodsType goodsType, int32_t maxProduceNum) = 0;
     virtual bool UplevelTech(THUAI9::TechType techType) = 0;
+    virtual std::string AskAI(int64_t currentGameTime, std::string prompt, std::string apiKey) = 0;
 };
 
 class IAPI
@@ -93,6 +94,7 @@ class IAPI
     // 等待下一帧
     virtual bool Wait() = 0;
     virtual std::future<bool> EndAllAction() = 0;
+    virtual std::future<std::string> AskAI(int64_t currentGameTime, std::string prompt, std::string apiKey = "") = 0;
     [[nodiscard]] virtual std::vector<std::shared_ptr<const THUAI9::Character>> GetCharacters() const = 0;
     [[nodiscard]] virtual std::vector<std::shared_ptr<const THUAI9::Character>> GetEnemyCharacters() const = 0;
     [[nodiscard]] virtual std::vector<std::vector<THUAI9::PlaceType>> GetFullMap() const = 0;
@@ -190,6 +192,7 @@ class CharacterAPI : public ICharacterAPI, public IGameTimer
     [[nodiscard]] int32_t GetFrameCount() const override;
     bool Wait() override;
     std::future<bool> EndAllAction() override;
+    std::future<std::string> AskAI(int64_t currentGameTime, std::string prompt, std::string apiKey = "") override;
 
     std::future<bool> Move(int64_t moveTimeInMilliseconds, double angle) override;
     std::future<bool> MoveRight(int64_t timeInMilliseconds) override;
@@ -258,6 +261,7 @@ class TeamAPI : public ITeamAPI, public IGameTimer
     [[nodiscard]] int32_t GetFrameCount() const override;
     bool Wait() override;
     std::future<bool> EndAllAction() override;
+    std::future<std::string> AskAI(int64_t currentGameTime, std::string prompt, std::string apiKey = "") override;
 
     [[nodiscard]] std::vector<std::shared_ptr<const THUAI9::Character>> GetCharacters() const override;
     [[nodiscard]] std::vector<std::shared_ptr<const THUAI9::Character>> GetEnemyCharacters() const override;
@@ -307,6 +311,7 @@ class CharacterDebugAPI : public ICharacterAPI, public IGameTimer
     bool Wait() override;
     [[nodiscard]] int32_t GetFrameCount() const override;
     std::future<bool> EndAllAction() override;
+    std::future<std::string> AskAI(int64_t currentGameTime, std::string prompt, std::string apiKey = "") override;
 
     std::future<bool> Move(int64_t moveTimeInMilliseconds, double angle) override;
     std::future<bool> MoveRight(int64_t timeInMilliseconds) override;
@@ -366,6 +371,7 @@ class TeamDebugAPI : public ITeamAPI, public IGameTimer
     [[nodiscard]] int32_t GetFrameCount() const override;
     bool Wait() override;
     std::future<bool> EndAllAction() override;
+    std::future<std::string> AskAI(int64_t currentGameTime, std::string prompt, std::string apiKey = "") override;
 
     [[nodiscard]] std::vector<std::shared_ptr<const THUAI9::Character>> GetCharacters() const override;
     [[nodiscard]] std::vector<std::shared_ptr<const THUAI9::Character>> GetEnemyCharacters() const override;

CAPI/cpp/API/include/Communication.h

@@ -50,6 +50,8 @@ class Communication
     bool ProduceGoods(int64_t teamID, THUAI9::GoodsType goodsType, int32_t maxProduceNum);
     bool UplevelTech(int64_t teamID, THUAI9::TechType techType);
 
+    std::string AskAI(int64_t teamID, int64_t currentGameTime, const std::string& prompt, const std::string& apiKey);
+
 private:
     std::unique_ptr<protobuf::AvailableService::Stub> THUAI9Stub;
     bool haveNewMessage = false;

CAPI/cpp/API/include/logic.h

@@ -136,6 +136,7 @@ class Logic : public ILogic
     bool BuildCharacter(THUAI9::CharacterType CharacterType, int32_t playerID);
     bool ProduceGoods(THUAI9::GoodsType goodsType, int32_t maxProduceNum);
     bool UplevelTech(THUAI9::TechType techType);
+    std::string AskAI(int64_t currentGameTime, std::string prompt, std::string apiKey);
 
     bool TryConnection();
     void ProcessMessage();

CAPI/cpp/API/src/AI.cpp

@@ -18,5 +18,6 @@ void AI::play(ICharacterAPI& api)
 void AI::play(ITeamAPI& api)
 {
     (void)api;
+    // auto answer = api.AskAI(/*currentGameTime*/, "your prompt", "<your uuid you can copy from eesast>").get();
     // TODO: implement team strategy.
 }

CAPI/cpp/API/src/API.cpp

@@ -81,12 +81,24 @@ std::future<bool> CharacterAPI::EndAllAction()
                       { return logic.EndAllAction(); });
 }
 
+std::future<std::string> CharacterAPI::AskAI(int64_t currentGameTime, std::string prompt, std::string apiKey)
+{
+    return std::async(std::launch::async, [this, currentGameTime, prompt = std::move(prompt), apiKey = std::move(apiKey)]() mutable
+                      { return logic.AskAI(currentGameTime, std::move(prompt), std::move(apiKey)); });
+}
+
 std::future<bool> TeamAPI::EndAllAction()
 {
     return std::async(std::launch::async, [this]()
                       { return logic.EndAllAction(); });
 }
 
+std::future<std::string> TeamAPI::AskAI(int64_t currentGameTime, std::string prompt, std::string apiKey)
+{
+    return std::async(std::launch::async, [this, currentGameTime, prompt = std::move(prompt), apiKey = std::move(apiKey)]() mutable
+                      { return logic.AskAI(currentGameTime, std::move(prompt), std::move(apiKey)); });
+}
+
 std::vector<std::shared_ptr<const THUAI9::Character>> CharacterAPI::GetCharacters() const
 {
     return logic.GetCharacters();

CAPI/cpp/API/src/Communication.cpp

@@ -187,6 +187,23 @@ bool Communication::UplevelTech(int64_t teamID, THUAI9::TechType techType)
     return status.ok() && result.act_success();
 }
 
+std::string Communication::AskAI(int64_t teamID, int64_t currentGameTime, const std::string& prompt, const std::string& apiKey)
+{
+    if (!ConsumeQuota(mtxLimit, counter, limit))
+        return {};
+
+    protobuf::StrategicAIResponse reply;
+    ClientContext context;
+    protobuf::StrategicAIRequest request;
+    request.set_team_id(teamID);
+    request.set_current_game_time(currentGameTime);
+    request.set_prompt(apiKey + "||" + prompt);
+    auto status = THUAI9Stub->AskAI(&context, request, &reply);
+    if (status.ok() && reply.act_success())
+        return reply.answer();
+    return {};
+}
+
 bool Communication::TryConnection(int32_t playerID, int32_t teamID)
 {
     protobuf::BoolRes reply;

CAPI/cpp/API/src/DebugAPI.cpp

@@ -99,6 +99,12 @@ std::future<bool> CharacterDebugAPI::EndAllAction()
                       { return logic.EndAllAction(); });
 }
 
+std::future<std::string> CharacterDebugAPI::AskAI(int64_t currentGameTime, std::string prompt, std::string apiKey)
+{
+    return std::async(std::launch::async, [this, currentGameTime, prompt = std::move(prompt), apiKey = std::move(apiKey)]() mutable
+                      { return logic.AskAI(currentGameTime, std::move(prompt), std::move(apiKey)); });
+}
+
 std::future<bool> CharacterDebugAPI::Move(int64_t moveTimeInMilliseconds, double angle)
 {
     logger->info("Move {} ms", moveTimeInMilliseconds);
@@ -352,6 +358,12 @@ std::future<bool> TeamDebugAPI::EndAllAction()
                       { return logic.EndAllAction(); });
 }
 
+std::future<std::string> TeamDebugAPI::AskAI(int64_t currentGameTime, std::string prompt, std::string apiKey)
+{
+    return std::async(std::launch::async, [this, currentGameTime, prompt = std::move(prompt), apiKey = std::move(apiKey)]() mutable
+                      { return logic.AskAI(currentGameTime, std::move(prompt), std::move(apiKey)); });
+}
+
 std::vector<std::shared_ptr<const THUAI9::Character>> TeamDebugAPI::GetCharacters() const
 {
     return logic.GetCharacters();

CAPI/cpp/API/src/logic.cpp

@@ -280,6 +280,11 @@ bool Logic::UplevelTech(THUAI9::TechType techType)
     return pComm->UplevelTech(teamID, techType);
 }
 
+std::string Logic::AskAI(int64_t currentGameTime, std::string prompt, std::string apiKey)
+{
+    return pComm->AskAI(teamID, currentGameTime, prompt, apiKey);
+}
+
 bool Logic::TryConnection()
 {
     return pComm->TryConnection(playerID, teamID);