mirror of
https://github.com/aljazceru/Auto-GPT.git
synced 2025-12-19 06:54:22 +01:00
bfbe613960
Additional changes: * Improve typing * Modularize message history memory & fix/refactor lots of things * Fix summarization * Move memory relevance calculation to MemoryItem & improve test * Fix import warnings in web_selenium.py * Remove `memory_add` ghost command * Implement overlap in `split_text` * Move memory tests into subdirectory * Remove deprecated `get_ada_embedding()` and helpers * Fix used token calculation in `chat_with_ai` * Replace Message TypedDict by dataclass * Fix AgentManager singleton issues in tests --------- Co-authored-by: Auto-GPT-Bot <github-bot@agpt.co>
55 lines
1.6 KiB
Python
55 lines
1.6 KiB
Python
import os
|
|
|
|
import openai.api_requestor
|
|
import pytest
|
|
from pytest_mock import MockerFixture
|
|
|
|
from tests.conftest import PROXY
|
|
from tests.vcr.vcr_filter import before_record_request, before_record_response
|
|
|
|
|
|
@pytest.fixture(scope="session")
|
|
def vcr_config():
|
|
# this fixture is called by the pytest-recording vcr decorator.
|
|
return {
|
|
"record_mode": "new_episodes",
|
|
"before_record_request": before_record_request,
|
|
"before_record_response": before_record_response,
|
|
"filter_headers": [
|
|
"Authorization",
|
|
"X-OpenAI-Client-User-Agent",
|
|
"User-Agent",
|
|
],
|
|
"match_on": ["method", "body"],
|
|
}
|
|
|
|
|
|
def patch_api_base(requestor):
|
|
new_api_base = f"{PROXY}/v1"
|
|
requestor.api_base = new_api_base
|
|
return requestor
|
|
|
|
|
|
@pytest.fixture
|
|
def patched_api_requestor(mocker: MockerFixture):
|
|
original_init = openai.api_requestor.APIRequestor.__init__
|
|
original_validate_headers = openai.api_requestor.APIRequestor._validate_headers
|
|
|
|
def patched_init(requestor, *args, **kwargs):
|
|
original_init(requestor, *args, **kwargs)
|
|
patch_api_base(requestor)
|
|
|
|
def patched_validate_headers(self, supplied_headers):
|
|
headers = original_validate_headers(self, supplied_headers)
|
|
headers["AGENT-MODE"] = os.environ.get("AGENT_MODE")
|
|
headers["AGENT-TYPE"] = os.environ.get("AGENT_TYPE")
|
|
return headers
|
|
|
|
if PROXY:
|
|
mocker.patch("openai.api_requestor.APIRequestor.__init__", new=patched_init)
|
|
mocker.patch.object(
|
|
openai.api_requestor.APIRequestor,
|
|
"_validate_headers",
|
|
new=patched_validate_headers,
|
|
)
|