Skip to content

Commit a1b6fe2

Browse files
committed
Install Flink agents for the default Python.
1 parent 87d8501 commit a1b6fe2

File tree

2 files changed

+4
-7
lines changed

2 files changed

+4
-7
lines changed

.github/workflows/ci.yml

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -144,7 +144,10 @@ jobs:
144144
with:
145145
version: "latest"
146146
- name: Install flink-agents
147-
run: bash tools/build.sh
147+
run: |
148+
bash tools/build.sh
149+
cd python
150+
pip install dist/*.whl
148151
- name: Install ollama
149152
run: bash tools/start_ollama_server.sh
150153
- name: Run Java IT

api/src/main/java/org/apache/flink/agents/api/chat/messages/ChatMessage.java

Lines changed: 0 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -32,9 +32,6 @@
3232
*/
3333
public class ChatMessage {
3434

35-
/** The key for the message type in the metadata. */
36-
public static final String MESSAGE_TYPE = "messageType";
37-
3835
private MessageRole role;
3936
private String content;
4037
private List<Map<String, Object>> toolCalls;
@@ -68,7 +65,6 @@ public ChatMessage(
6865
this.content = content != null ? content : "";
6966
this.toolCalls = toolCalls != null ? toolCalls : new ArrayList<>();
7067
this.extraArgs = extraArgs != null ? new HashMap<>(extraArgs) : new HashMap<>();
71-
this.extraArgs.put(MESSAGE_TYPE, this.role);
7268
}
7369

7470
public MessageRole getRole() {
@@ -77,7 +73,6 @@ public MessageRole getRole() {
7773

7874
public void setRole(MessageRole role) {
7975
this.role = role;
80-
this.extraArgs.put(MESSAGE_TYPE, this.role);
8176
}
8277

8378
public String getContent() {
@@ -102,7 +97,6 @@ public Map<String, Object> getExtraArgs() {
10297

10398
public void setExtraArgs(Map<String, Object> extraArgs) {
10499
this.extraArgs = extraArgs != null ? extraArgs : new HashMap<>();
105-
this.extraArgs.put(MESSAGE_TYPE, this.role);
106100
}
107101

108102
@JsonIgnore

0 commit comments

Comments
 (0)