Skip to content

Commit 0be86a4

Browse files
committed
Fix attention test on Windows.
1 parent ed2c475 commit 0be86a4

File tree

2 files changed

+7
-12
lines changed

2 files changed

+7
-12
lines changed

tests/utils.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@
1717
import asyncio
1818
import json
1919
import sys
20+
from datetime import datetime, timedelta, timezone
2021
from typing import Any, Dict, Iterable, List, Mapping, Optional, Union
2122

2223
from langchain.callbacks.manager import (
@@ -171,6 +172,12 @@ def user(self, msg: Union[str, dict]):
171172
final_transcript=msg,
172173
action_uid=uid,
173174
is_success=True,
175+
event_created_at=(
176+
datetime.now(timezone.utc) + timedelta(milliseconds=1)
177+
).isoformat(),
178+
action_finished_at=(
179+
datetime.now(timezone.utc) + timedelta(milliseconds=1)
180+
).isoformat(),
174181
),
175182
]
176183
)

tests/v2_x/test_attention_library.py

Lines changed: 0 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -97,18 +97,6 @@ def test_1_3(config_1):
9797
chat << "got inattentive"
9898

9999

100-
def test_1_3(config_1):
101-
chat = TestChat(
102-
config_1,
103-
llm_completions=[],
104-
)
105-
106-
chat >> "down"
107-
chat << "down"
108-
chat >> "hello there"
109-
chat << "got inattentive"
110-
111-
112100
def test_1_4(config_1):
113101
chat = TestChat(
114102
config_1,

0 commit comments

Comments
 (0)