Austin/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2024-11-16 11:42:29 +08:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

improve: CI experience (#5168)

Browse Source
This commit is contained in:
Yeuoly 2024-06-13 23:16:28 +08:00 committed by GitHub
parent 0cccf9c67d
commit e8afc416dd
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 2 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/tests/integration_tests/model_runtime/__mock/anthropic.py
View File

@ -1,6 +1,5 @@
import os
from collections.abc import Iterable
from time import sleep
from typing import Any, Literal, Union
import anthropic
@ -64,7 +63,6 @@ class MockAnthropicClass:
index = 0
for i in range(0, len(full_response_text)):
sleep(0.1)
yield ContentBlockDeltaEvent(
type='content_block_delta',
delta=TextDelta(text=full_response_text[i], type='text_delta'),

4
api/tests/integration_tests/model_runtime/__mock/openai_chat.py
View File

@ -1,7 +1,7 @@
import re
from collections.abc import Generator
from json import dumps, loads
from time import sleep, time
from time import time
# import monkeypatch
from typing import Any, Literal, Optional, Union
@ -112,7 +112,6 @@ class MockChatClass:
if not function_call:
tool_calls = MockChatClass.generate_tool_calls(tools=tools)
sleep(1)
return _ChatCompletion(
id='cmpl-3QJQa5jXJ5Z5X',
choices=[
@ -151,7 +150,6 @@ class MockChatClass:
full_text = "Hello, world!\n\n```python\nprint('Hello, world!')\n```"
for i in range(0, len(full_text) + 1):
sleep(0.1)
if i == len(full_text):
yield ChatCompletionChunk(
id='cmpl-3QJQa5jXJ5Z5X',

4
api/tests/integration_tests/model_runtime/__mock/openai_completion.py
View File

@ -1,6 +1,6 @@
import re
from collections.abc import Generator
from time import sleep, time
from time import time
# import monkeypatch
from typing import Any, Literal, Optional, Union
@ -20,7 +20,6 @@ class MockCompletionsClass:
def mocked_openai_completion_create_sync(
model: str
) -> CompletionMessage:
sleep(1)
return CompletionMessage(
id="cmpl-3QJQa5jXJ5Z5X",
object="text_completion",
@ -48,7 +47,6 @@ class MockCompletionsClass:
) -> Generator[CompletionMessage, None, None]:
full_text = "Hello, world!\n\n```python\nprint('Hello, world!')\n```"
for i in range(0, len(full_text) + 1):
sleep(0.1)
if i == len(full_text):
yield CompletionMessage(
id="cmpl-3QJQa5jXJ5Z5X",