Merge pull request #44 from atomiechen/ft_import

Feature: improve imports for better IDE support

.gitignore

@@ -2,6 +2,7 @@
 .DS_Store
 
 dist
+build
 *.egg-info
 
 # credentials

pyproject.toml

@@ -41,6 +41,6 @@ addopts = [
 ]
 
 [tool.ruff.lint.per-file-ignores]
-# ref: https://github.com/astral-sh/ruff/issues/2407#issuecomment-1974783543
-# Ignore unused imports and import * in __init__.py files
-"__init__.py" = ["F401", "F403"]
+# # ref: https://github.com/astral-sh/ruff/issues/2407#issuecomment-1974783543
+# # Ignore unused imports and import * in __init__.py files
+# "__init__.py" = ["F401", "F403"]

src/handyllm/__init__.py

@@ -1,10 +1,38 @@
-from .openai_client import *
-from .requestor import *
-from .openai_api import *
-from .endpoint_manager import *
-from .prompt_converter import *
-from .utils import *
-from .types import *
-from .hprompt import *
-from .cache_manager import *
-from .response import *
+from .openai_client import OpenAIClient as OpenAIClient, ClientMode as ClientMode
+from .requestor import (
+    Requestor as Requestor,
+    DictRequestor as DictRequestor,
+    BinRequestor as BinRequestor,
+    ChatRequestor as ChatRequestor,
+    CompletionsRequestor as CompletionsRequestor,
+)
+from .openai_api import OpenAIAPI as OpenAIAPI
+from .endpoint_manager import EndpointManager as EndpointManager, Endpoint as Endpoint
+from .prompt_converter import PromptConverter as PromptConverter
+from .utils import (
+    stream_chat_all as stream_chat_all,
+    stream_chat as stream_chat,
+    stream_completions as stream_completions,
+    astream_chat_all as astream_chat_all,
+    astream_chat as astream_chat,
+    astream_completions as astream_completions,
+    stream_to_file as stream_to_file,
+    astream_to_file as astream_to_file,
+    VM as VM,
+)
+from .hprompt import (
+    HandyPrompt as HandyPrompt,
+    ChatPrompt as ChatPrompt,
+    CompletionsPrompt as CompletionsPrompt,
+    loads as loads,
+    load as load,
+    load_from as load_from,
+    dumps as dumps,
+    dump as dump,
+    dump_to as dump_to,
+    load_var_map as load_var_map,
+    RunConfig as RunConfig,
+    RecordRequestMode as RecordRequestMode,
+    CredentialType as CredentialType,
+)
+from .cache_manager import CacheManager as CacheManager

src/handyllm/response.py

@@ -9,6 +9,15 @@
     "CompletionsResponse",
     "CompletionsChunkChoice",
     "CompletionsChunk",
+    "Function",
+    "ToolCall",
+    "TopLogProbItem",
+    "LogProbItem",
+    "Logprobs",
+    "Usage",
+    "ToolCallDelta",
+    "ChatChunkDelta",
+    "CompletionLogprobs",
 ]
 
 from typing import List, MutableMapping, Optional, Sequence, TypedDict