Skip to content

Commit

Permalink
Refresh EndpointConfig request headers (#12904)
Browse files Browse the repository at this point in the history
* add debug log

* update headers

* make dev release

* fix numpy import error

* fix typing imports

* add changelog, cleanup
  • Loading branch information
ancalita committed Oct 10, 2023
1 parent d39fd35 commit f155c6c
Show file tree
Hide file tree
Showing 3 changed files with 9 additions and 5 deletions.
1 change: 1 addition & 0 deletions changelog/12904.bugfix.md
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
Refresh headers used in requests (e.g. action server requests) made by `EndpointConfig` using its `headers` attribute.
10 changes: 5 additions & 5 deletions rasa/plugin.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,15 +6,15 @@
import pluggy

from rasa.cli import SubParsersAction
from rasa.engine.storage.storage import ModelMetadata
from rasa.shared.core.trackers import DialogueStateTracker
from rasa.shared.nlu.training_data.message import Message

if TYPE_CHECKING:
from rasa.core.brokers.broker import EventBroker
from rasa.core.tracker_store import TrackerStore
from rasa.engine.graph import SchemaNode
from rasa.engine.storage.storage import ModelMetadata
from rasa.shared.core.domain import Domain
from rasa.shared.core.trackers import DialogueStateTracker
from rasa.shared.nlu.training_data.message import Message
from rasa.utils.endpoints import EndpointConfig


Expand Down Expand Up @@ -95,8 +95,8 @@ def init_telemetry(endpoints_file: Optional[Text]) -> None:

@hookspec # type: ignore[misc]
def mock_tracker_for_evaluation(
example: Message, model_metadata: Optional[ModelMetadata]
) -> Optional[DialogueStateTracker]:
example: "Message", model_metadata: Optional["ModelMetadata"]
) -> Optional["DialogueStateTracker"]:
"""Generate a mocked tracker for NLU evaluation."""


Expand Down
3 changes: 3 additions & 0 deletions rasa/utils/endpoints.py
Original file line number Diff line number Diff line change
Expand Up @@ -149,6 +149,9 @@ async def request(
headers.update(kwargs["headers"])
del kwargs["headers"]

if self.headers:
headers.update(self.headers)

url = concat_url(self.url, subpath)

sslcontext = None
Expand Down

0 comments on commit f155c6c

Please sign in to comment.