Skip to content

Commit 93d7093

Browse files
MarcMcIntoshMarcMcIntosh
committed
chore(chat): move cmessages stub to fixtures.
1 parent 07bd239 commit 93d7093

File tree

3 files changed

+123
-109
lines changed

3 files changed

+123
-109
lines changed

‎refact-agent/gui/src/__fixtures__/chat_db_cmessages.ts‎

Lines changed: 120 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,120 @@
1+
import { CMessage, CMessageFromChatDB } from "../services/refact/types";
2+
3+
export const CMESSAGES_STUB: CMessage[] = [
4+
{
5+
cmessage_belongs_to_cthread_id: "test13thread1739988322_2",
6+
cmessage_alt: 0,
7+
cmessage_num: 0,
8+
cmessage_prev_alt: -1,
9+
cmessage_usage_model: "",
10+
cmessage_usage_prompt: 0,
11+
cmessage_usage_completion: 0,
12+
cmessage_json: {
13+
role: "system",
14+
content: "You answer only with jokes.",
15+
},
16+
},
17+
{
18+
cmessage_belongs_to_cthread_id: "test13thread1739988322_2",
19+
cmessage_alt: 0,
20+
cmessage_num: 1,
21+
cmessage_prev_alt: 0,
22+
cmessage_usage_model: "",
23+
cmessage_usage_prompt: 0,
24+
cmessage_usage_completion: 0,
25+
cmessage_json: {
26+
role: "user",
27+
content: "Hello mister assistant, I have a question for you",
28+
},
29+
},
30+
{
31+
cmessage_belongs_to_cthread_id: "test13thread1739988322_2",
32+
cmessage_alt: 1,
33+
cmessage_num: 1,
34+
cmessage_prev_alt: 0,
35+
cmessage_usage_model: "",
36+
cmessage_usage_prompt: 0,
37+
cmessage_usage_completion: 0,
38+
cmessage_json: {
39+
role: "user",
40+
content: "Find Frog in this project",
41+
},
42+
},
43+
{
44+
cmessage_belongs_to_cthread_id: "test13thread1739988322_2",
45+
cmessage_alt: 0,
46+
cmessage_num: 2,
47+
cmessage_prev_alt: 0,
48+
cmessage_usage_model: "gpt-4o-mini",
49+
cmessage_usage_prompt: 0,
50+
cmessage_usage_completion: 0,
51+
cmessage_json: {
52+
role: "assistant",
53+
content: "yes?.",
54+
},
55+
},
56+
{
57+
cmessage_belongs_to_cthread_id: "test13thread1739988322_2",
58+
cmessage_alt: 1,
59+
cmessage_num: 2,
60+
cmessage_prev_alt: 1,
61+
cmessage_usage_model: "gpt-4o-mini",
62+
cmessage_usage_prompt: 0,
63+
cmessage_usage_completion: 0,
64+
cmessage_json: {
65+
role: "assistant",
66+
content: "Birds aren't real",
67+
},
68+
},
69+
{
70+
cmessage_belongs_to_cthread_id: "test13thread1739988322_2",
71+
cmessage_alt: 0,
72+
cmessage_num: 3,
73+
cmessage_prev_alt: 0,
74+
cmessage_usage_model: "gpt-4o-mini",
75+
cmessage_usage_prompt: 0,
76+
cmessage_usage_completion: 0,
77+
cmessage_json: {
78+
role: "user",
79+
content: "Find Frog in this project",
80+
},
81+
},
82+
{
83+
cmessage_belongs_to_cthread_id: "test13thread1739988322_2",
84+
cmessage_alt: 0,
85+
cmessage_num: 4,
86+
cmessage_prev_alt: 0,
87+
cmessage_usage_model: "gpt-4o-mini",
88+
cmessage_usage_prompt: 1210,
89+
cmessage_usage_completion: 15,
90+
cmessage_json: {
91+
role: "assistant",
92+
content: "",
93+
tool_calls: [
94+
{
95+
index: 0,
96+
id: "call_8PSEh32Hhivfdxc50XKNwW8y",
97+
function: {
98+
arguments: '{"symbol":"Frog"}',
99+
name: "references",
100+
},
101+
type: "function",
102+
},
103+
],
104+
// usage: {
105+
// prompt_tokens: 1210,
106+
// completion_tokens: 15,
107+
// total_tokens: 1225,
108+
// },
109+
},
110+
},
111+
];
112+
113+
export const CSMESSAGES_NETWORK_STUB: CMessageFromChatDB[] = CMESSAGES_STUB.map(
114+
(cmessage) => {
115+
return {
116+
...cmessage,
117+
cmessage_json: JSON.stringify(cmessage.cmessage_json),
118+
};
119+
},
120+
);

‎refact-agent/gui/src/__fixtures__/index.ts‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,3 +8,4 @@ export * from "./survey_questions";
88
export * from "./chat_links_response";
99
export * from "./chat_config_thread";
1010
export * from "./chat_textdoc";
11+
export * from "./chat_db_cmessages";

‎refact-agent/gui/src/features/ChatDB/makeMessageTree.test.ts‎

Lines changed: 2 additions & 109 deletions
Original file line numberDiff line numberDiff line change
@@ -2,116 +2,9 @@ import { expect, describe, test } from "vitest";
22
import { CMessage } from "../../services/refact";
33
import { CMessageNode } from "./chatDbMessagesSlice";
44
import { makeMessageTree } from "./makeMessageTrie";
5+
import { CMESSAGES_STUB } from "../../__fixtures__";
56

6-
const STUB: CMessage[] = [
7-
{
8-
cmessage_belongs_to_cthread_id: "test13thread1739988322_2",
9-
cmessage_alt: 0,
10-
cmessage_num: 0,
11-
cmessage_prev_alt: -1,
12-
cmessage_usage_model: "",
13-
cmessage_usage_prompt: 0,
14-
cmessage_usage_completion: 0,
15-
cmessage_json: {
16-
role: "system",
17-
content: "You answer only with jokes.",
18-
},
19-
},
20-
{
21-
cmessage_belongs_to_cthread_id: "test13thread1739988322_2",
22-
cmessage_alt: 0,
23-
cmessage_num: 1,
24-
cmessage_prev_alt: 0,
25-
cmessage_usage_model: "",
26-
cmessage_usage_prompt: 0,
27-
cmessage_usage_completion: 0,
28-
cmessage_json: {
29-
role: "user",
30-
content: "Hello mister assistant, I have a question for you",
31-
},
32-
},
33-
{
34-
cmessage_belongs_to_cthread_id: "test13thread1739988322_2",
35-
cmessage_alt: 1,
36-
cmessage_num: 1,
37-
cmessage_prev_alt: 0,
38-
cmessage_usage_model: "",
39-
cmessage_usage_prompt: 0,
40-
cmessage_usage_completion: 0,
41-
cmessage_json: {
42-
role: "user",
43-
content: "Find Frog in this project",
44-
},
45-
},
46-
{
47-
cmessage_belongs_to_cthread_id: "test13thread1739988322_2",
48-
cmessage_alt: 0,
49-
cmessage_num: 2,
50-
cmessage_prev_alt: 0,
51-
cmessage_usage_model: "gpt-4o-mini",
52-
cmessage_usage_prompt: 0,
53-
cmessage_usage_completion: 0,
54-
cmessage_json: {
55-
role: "assistant",
56-
content: "yes?.",
57-
},
58-
},
59-
{
60-
cmessage_belongs_to_cthread_id: "test13thread1739988322_2",
61-
cmessage_alt: 1,
62-
cmessage_num: 2,
63-
cmessage_prev_alt: 1,
64-
cmessage_usage_model: "gpt-4o-mini",
65-
cmessage_usage_prompt: 0,
66-
cmessage_usage_completion: 0,
67-
cmessage_json: {
68-
role: "assistant",
69-
content: "Birds aren't real",
70-
},
71-
},
72-
{
73-
cmessage_belongs_to_cthread_id: "test13thread1739988322_2",
74-
cmessage_alt: 0,
75-
cmessage_num: 3,
76-
cmessage_prev_alt: 0,
77-
cmessage_usage_model: "gpt-4o-mini",
78-
cmessage_usage_prompt: 0,
79-
cmessage_usage_completion: 0,
80-
cmessage_json: {
81-
role: "user",
82-
content: "Find Frog in this project",
83-
},
84-
},
85-
{
86-
cmessage_belongs_to_cthread_id: "test13thread1739988322_2",
87-
cmessage_alt: 0,
88-
cmessage_num: 4,
89-
cmessage_prev_alt: 0,
90-
cmessage_usage_model: "gpt-4o-mini",
91-
cmessage_usage_prompt: 1210,
92-
cmessage_usage_completion: 15,
93-
cmessage_json: {
94-
role: "assistant",
95-
content: "",
96-
tool_calls: [
97-
{
98-
index: 0,
99-
id: "call_8PSEh32Hhivfdxc50XKNwW8y",
100-
function: {
101-
arguments: '{"symbol":"Frog"}',
102-
name: "references",
103-
},
104-
type: "function",
105-
},
106-
],
107-
// usage: {
108-
// prompt_tokens: 1210,
109-
// completion_tokens: 15,
110-
// total_tokens: 1225,
111-
// },
112-
},
113-
},
114-
];
7+
const STUB = CMESSAGES_STUB;
1158

1169
describe("makeMessageTree", () => {
11710
test("no root", () => {

0 commit comments

Comments
 (0)