Cleaned up and organized imports

This commit is contained in:
David J. Bianco
2024-08-23 11:43:15 -04:00
parent a180bb58a2
commit b49e743e7c

16
ssh_server.py Normal file → Executable file
View File

@ -1,32 +1,26 @@
#!/usr/bin/env python3 #!/usr/bin/env python3
from configparser import ConfigParser
import asyncio import asyncio
import asyncssh import asyncssh
import sys import sys
import json import json
from typing import Optional from typing import Optional
import logging import logging
import datetime import datetime
import uuid import uuid
from base64 import b64encode from base64 import b64encode
from operator import itemgetter
from configparser import ConfigParser
from langchain_openai import ChatOpenAI from langchain_openai import ChatOpenAI
from langchain_aws import ChatBedrock, ChatBedrockConverse from langchain_aws import ChatBedrock, ChatBedrockConverse
from langchain_google_genai import ChatGoogleGenerativeAI from langchain_google_genai import ChatGoogleGenerativeAI
from langchain_core.messages import HumanMessage, SystemMessage, trim_messages
from langchain_core.messages import HumanMessage
from langchain_core.chat_history import BaseChatMessageHistory, InMemoryChatMessageHistory from langchain_core.chat_history import BaseChatMessageHistory, InMemoryChatMessageHistory
from langchain_core.runnables.history import RunnableWithMessageHistory from langchain_core.runnables.history import RunnableWithMessageHistory
from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder from langchain_core.prompts import ChatPromptTemplate, MessagesPlaceholder
from langchain_core.messages import SystemMessage, trim_messages
from langchain_core.runnables import RunnablePassthrough from langchain_core.runnables import RunnablePassthrough
from operator import itemgetter
from configparser import ConfigParser
async def handle_client(process: asyncssh.SSHServerProcess) -> None: async def handle_client(process: asyncssh.SSHServerProcess) -> None:
# This is the main loop for handling SSH client connections. # This is the main loop for handling SSH client connections.
# Any user interaction should be done here. # Any user interaction should be done here.