diff --git a/trustgraph-base/trustgraph/base/__init__.py b/trustgraph-base/trustgraph/base/__init__.py index 5fad114f..d5e539c6 100644 --- a/trustgraph-base/trustgraph/base/__init__.py +++ b/trustgraph-base/trustgraph/base/__init__.py @@ -15,6 +15,6 @@ from . request_response_spec import RequestResponseSpec from . llm_service import LlmService, LlmResult from . embeddings_service import EmbeddingsService from . embeddings_client import EmbeddingsClientSpec -from . text_completion_client import TextCompletionsClientSpec +from . text_completion_client import TextCompletionClientSpec from . prompt_client import PromptClientSpec diff --git a/trustgraph-base/trustgraph/base/embeddings_client.py b/trustgraph-base/trustgraph/base/embeddings_client.py index 89bdfa86..e62a8d79 100644 --- a/trustgraph-base/trustgraph/base/embeddings_client.py +++ b/trustgraph-base/trustgraph/base/embeddings_client.py @@ -1,6 +1,6 @@ from . request_response_spec import RequestResponse, RequestResponseSpec -from .. schema import EmbeddingsReqeust, EmbeddingsResponse +from .. schema import EmbeddingsRequest, EmbeddingsResponse class EmbeddingsClient(RequestResponse): async def embed(self, text, timeout=30): diff --git a/trustgraph-base/trustgraph/base/prompt_client.py b/trustgraph-base/trustgraph/base/prompt_client.py index 393b3b5f..74cec56b 100644 --- a/trustgraph-base/trustgraph/base/prompt_client.py +++ b/trustgraph-base/trustgraph/base/prompt_client.py @@ -2,7 +2,7 @@ import json from . request_response_spec import RequestResponse, RequestResponseSpec -from .. schema import EmbeddingsReqeust, EmbeddingsResponse +from .. schema import PromptRequest, PromptResponse class PromptClient(RequestResponse): @@ -42,11 +42,11 @@ class PromptClientSpec(RequestResponseSpec): def __init__( self, request_name, response_name, ): - super(EmbeddingsRequestResponseSpec, self).__init__( + super(PromptClientSpec, self).__init__( request_name = request_name, - request_schema = EmbeddingsRequest, + request_schema = PromptRequest, response_name = response_name, - response_schema = EmbeddingsResponse, - impl = TextCompletionClient, + response_schema = PromptResponse, + impl = PromptClient, ) diff --git a/trustgraph-base/trustgraph/base/text_completion_client.py b/trustgraph-base/trustgraph/base/text_completion_client.py index 2926ca4c..aba2fada 100644 --- a/trustgraph-base/trustgraph/base/text_completion_client.py +++ b/trustgraph-base/trustgraph/base/text_completion_client.py @@ -1,6 +1,6 @@ from . request_response_spec import RequestResponse, RequestResponseSpec -from .. schema import EmbeddingsReqeust, EmbeddingsResponse +from .. schema import TextCompletionRequest, TextCompletionResponse class TextCompletionClient(RequestResponse): async def text_completion(self, system, prompt, timeout=600): @@ -20,11 +20,11 @@ class TextCompletionClientSpec(RequestResponseSpec): def __init__( self, request_name, response_name, ): - super(EmbeddingsRequestResponseSpec, self).__init__( + super(TextCompletionClientSpec, self).__init__( request_name = request_name, - request_schema = EmbeddingsRequest, + request_schema = TextCompletionRequest, response_name = response_name, - response_schema = EmbeddingsResponse, + response_schema = TextCompletionResponse, impl = TextCompletionClient, )