(update) Sync with internal (#5)

Co-authored-by: Aleksandr Liadov <[email protected]>
Stellantis-ADX · Dec 20, 2024 · d6cd376 · d6cd376
1 parent 36cf7fa
commit d6cd376
Show file tree

Hide file tree

Showing 34 changed files with 3,932 additions and 2,245 deletions.
diff --git a/.devcontainer/devcontainer.json b/.devcontainer/devcontainer.json
diff --git a/.github/dependabot.yml b/.github/dependabot.yml
diff --git a/.github/workflows/combine-prs.yml b/.github/workflows/combine-prs.yml
diff --git a/.github/workflows/openai-review.yml b/.github/workflows/openai-review.yml
diff --git a/.github/workflows/versioning.yml b/.github/workflows/versioning.yml
diff --git a/action.yml b/action.yml
@@ -187,11 +187,11 @@ inputs:
   heavy_model_port:
     required: false
     description: 'Port for the heavy model'
-    default: '44902'
+    default: '44951'
   model_temperature:
     required: false
     description: 'Temperature for language model'
-    default: '0.1'
+    default: '0.2'
   retries:
     required: false
     description:

diff --git a/core/bots/__init__.py b/core/bots/__init__.py
diff --git a/core/bot.py → core/bots/bot.py b/core/bot.py → core/bots/bot.py
@@ -14,21 +14,19 @@
 
 
 class Ids(BaseModel):
-    conversationId: Optional[str]
-    parentMessageId: Optional[str]
+    # Currently not used
+    conversationId: Optional[str] = None
+    parentMessageId: Optional[str] = None
 
     def __str__(self) -> str:
         return f"conversationId: {self.conversationId}, parentMessageId: {self.parentMessageId} \n"
 
 
 class AiResponse(BaseModel):
-    message: str
-    ids: Ids
+    message: str = ""
 
     def __str__(self) -> str:
-        return str(self.ids) + "\n".join(
-            [f"{line}" for line in self.message.split("\n")]
-        )
+        return "\n".join([f"{line}" for line in self.message.split("\n")])
 
 
 class ModelOptions(ABC):
@@ -47,5 +45,5 @@ def __init__(self, options: Options, model_options: ModelOptions):
         self.model_options = model_options
 
     @abstractmethod
-    def chat(self, message, ids):
+    def chat(self, message: str) -> AiResponse:
         pass