Skip to content

Commit

Permalink
yet another examples formatting fix
Browse files Browse the repository at this point in the history
  • Loading branch information
nizamovtimur committed Sep 7, 2024
1 parent a5470a8 commit 0d2992a
Show file tree
Hide file tree
Showing 3 changed files with 7 additions and 4 deletions.
4 changes: 2 additions & 2 deletions notebooks/llamator-api-example.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -154,8 +154,8 @@
"source": [
"from llamator.client.specific_chat_clients import ClientOpenAI\n",
"\n",
"api_key=\"lm-studio\"\n",
"base_url=\"http://localhost:1234/v1\"\n",
"api_key = \"lm-studio\"\n",
"base_url = \"http://localhost:1234/v1\"\n",
"model = \"model-identifier\"\n",
"\n",
"attack_model = ClientOpenAI(api_key=api_key,\n",
Expand Down
4 changes: 2 additions & 2 deletions notebooks/llamator-selenium-example.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -178,8 +178,8 @@
"source": [
"from llamator.client.specific_chat_clients import ClientOpenAI\n",
"\n",
"api_key=\"lm-studio\"\n",
"base_url=\"http://localhost:1234/v1\"\n",
"api_key = \"lm-studio\"\n",
"base_url = \"http://localhost:1234/v1\"\n",
"model = \"model-identifier\"\n",
"\n",
"attack_model = ClientOpenAI(api_key=api_key,\n",
Expand Down
3 changes: 3 additions & 0 deletions src/llamator/client/specific_chat_clients.py
Original file line number Diff line number Diff line change
Expand Up @@ -178,6 +178,9 @@ def __init__(
self.model_description = model_description


# TODO: вернуть конвертер


def interact(self, history: List[Dict[str, str]], messages: List[Dict[str, str]]) -> Dict[str, str]:
"""
Takes conversation history and new messages, sends a request to the OpenAI-compatible API, and returns the response.
Expand Down

0 comments on commit 0d2992a

Please sign in to comment.