apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-17 09:11:03 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Clients

Browse source
This commit is contained in:
Cyber MacGeddon 2025-04-19 13:54:42 +01:00
parent c4d961bdb7
commit c05ad5c215
4 changed files with 11 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

2
trustgraph-base/trustgraph/base/__init__.py
View file

@ -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

2
trustgraph-base/trustgraph/base/embeddings_client.py
View file

@ -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):

10
trustgraph-base/trustgraph/base/prompt_client.py
View file

@ -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,
)

8
trustgraph-base/trustgraph/base/text_completion_client.py
View file

@ -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,
)