feat(model): add ServerConfig and ClientConfig (#1266)
This commit is contained in:
parent
fbfc5cc96e
commit
165ff92df1
1 changed files with 331 additions and 303 deletions
|
|
@ -16,9 +16,9 @@ use futures::{StreamExt, stream::BoxStream};
|
|||
use http::{HeaderName, HeaderValue};
|
||||
use rmcp::{
|
||||
model::{
|
||||
CallToolRequestParams, CancelledNotificationParam, ClientJsonRpcMessage, ClientRequest,
|
||||
DiscoverResult, InitializeRequestParams, ProtocolVersion, Request, RequestId,
|
||||
RequestMetaObject, ServerJsonRpcMessage,
|
||||
CallToolRequestParams, CancelledNotificationParam, ClientConfig, ClientJsonRpcMessage,
|
||||
ClientRequest, DiscoverResult, InitializeRequestParams, ProtocolVersion, Request,
|
||||
RequestId, RequestMetaObject, ServerJsonRpcMessage,
|
||||
},
|
||||
service::{
|
||||
ClientLifecycleMode, PeerRequestOptions, RequestHandle, RoleClient, RunningService,
|
||||
|
|
@ -345,7 +345,7 @@ impl StreamableHttpClient for ScriptedClient {
|
|||
}
|
||||
|
||||
struct Harness {
|
||||
client: RunningService<RoleClient, InitializeRequestParams>,
|
||||
client: RunningService<RoleClient, ClientConfig>,
|
||||
started: mpsc::UnboundedReceiver<Posted>,
|
||||
controls: mpsc::UnboundedReceiver<ControlPost>,
|
||||
incoming: mpsc::UnboundedSender<Result<Sse, SseError>>,
|
||||
|
|
@ -402,8 +402,7 @@ impl Harness {
|
|||
config,
|
||||
);
|
||||
let client =
|
||||
serve_client_with_lifecycle(InitializeRequestParams::default(), transport, lifecycle)
|
||||
.await?;
|
||||
serve_client_with_lifecycle(ClientConfig::default(), transport, lifecycle).await?;
|
||||
Ok(Self {
|
||||
client,
|
||||
started: requests,
|
||||
|
|
|
|||
Loading…
Reference in a new issue