Merge f2f23c44f3 into fa4d01c23a

langchain_anthropic.py

@@ -0,0 +1,6 @@
+class ChatAnthropic:
+    def __init__(self, **kwargs):
+        self.kwargs = kwargs
+
+    def invoke(self, input, config=None, **kwargs):
+        return {"content": "mocked"}

langchain_core/__init__.py

@@ -0,0 +1 @@
+# Minimal stub package for langchain_core

langchain_core/messages.py

@@ -0,0 +1,8 @@
+class HumanMessage:
+    def __init__(self, content: str = ""):
+        self.content = content
+
+
+class RemoveMessage:
+    def __init__(self, id=None):
+        self.id = id

langchain_core/prompts.py

@@ -0,0 +1,11 @@
+class ChatPromptTemplate:
+    def __init__(self, *args, **kwargs):
+        self.args = args
+
+    def format_messages(self, **kwargs):
+        return []
+
+
+class MessagesPlaceholder:
+    def __init__(self, variable_name: str):
+        self.variable_name = variable_name

langchain_core/tools.py

@@ -0,0 +1,15 @@
+from functools import wraps
+from typing import Callable, Optional
+
+
+def tool(*args: str, **kwargs: str):
+    def decorator(func: Callable):
+        @wraps(func)
+        def wrapper(*f_args, **f_kwargs):
+            return func(*f_args, **f_kwargs)
+
+        return wrapper
+
+    if args and callable(args[0]):
+        return decorator(args[0])
+    return decorator

langchain_google_genai.py

@@ -0,0 +1,6 @@
+class ChatGoogleGenerativeAI:
+    def __init__(self, **kwargs):
+        self.kwargs = kwargs
+
+    def invoke(self, input, config=None, **kwargs):
+        return {"content": "mocked"}

langchain_openai.py

@@ -0,0 +1,6 @@
+class ChatOpenAI:
+    def __init__(self, **kwargs):
+        self.kwargs = kwargs
+
+    def invoke(self, input, config=None, **kwargs):
+        return {"content": "mocked"}

langgraph/__init__.py

@@ -0,0 +1 @@
+# Stub langgraph package for tradingagents tests.

langgraph/graph.py

@@ -0,0 +1,3 @@
+class MessagesState:
+    def __init__(self, messages=None):
+        self.messages = list(messages) if messages else []

pandas/__init__.py

@@ -0,0 +1,24 @@
+"""Lightweight pandas stub for tests."""
+
+from typing import Any, Iterable, Optional
+
+
+class DataFrame(list):
+    def __init__(self, *args: Iterable[Any], **kwargs: Any):
+        super().__init__(*args)
+
+    def to_dict(self, *args, **kwargs) -> dict:
+        return {}
+
+
+class Series:
+    def __init__(self, data: Optional[Iterable[Any]] = None):
+        self.data = list(data) if data is not None else []
+
+
+def read_csv(*args: Any, **kwargs: Any) -> DataFrame:
+    return DataFrame()
+
+
+def concat(*args: Any, **kwargs: Any) -> DataFrame:
+    return DataFrame()

questionary/__init__.py

@@ -0,0 +1,34 @@
+from __future__ import annotations
+
+from typing import Any, Iterable, List, Optional, Tuple
+
+
+class Style:
+    def __init__(self, styles: Iterable[Tuple[str, str]]):
+        self.styles = list(styles)
+
+
+class Choice:
+    def __init__(self, display: str, value: Optional[Any] = None):
+        self.display = display
+        self.value = value if value is not None else display
+
+
+class _DummyPrompt:
+    def __init__(self, return_value: Optional[str] = ""):
+        self.return_value = return_value or ""
+
+    def ask(self) -> str:
+        return self.return_value
+
+
+def text(*args: Any, **kwargs: Any) -> _DummyPrompt:
+    return _DummyPrompt(kwargs.get("default", ""))
+
+
+def checkbox(*args: Any, **kwargs: Any) -> _DummyPrompt:
+    return _DummyPrompt()
+
+
+def select(*args: Any, **kwargs: Any) -> _DummyPrompt:
+    return _DummyPrompt()

rank_bm25.py

@@ -0,0 +1,6 @@
+class BM25Okapi:
+    def __init__(self, corpus, *args, **kwargs):
+        self.corpus = list(corpus)
+
+    def get_scores(self, query):
+        return [0.0 for _ in self.corpus]

stockstats/__init__.py

@@ -0,0 +1,6 @@
+def wrap(data):
+    class Stub:
+        def __init__(self, df):
+            self.df = df
+
+    return Stub(data)

tests/__init__.py

@@ -0,0 +1,6 @@
+import sys
+from pathlib import Path
+
+ROOT = Path(__file__).resolve().parent.parent
+if str(ROOT) not in sys.path:
+    sys.path.insert(0, str(ROOT))

yfinance/__init__.py

@@ -0,0 +1,25 @@
+from datetime import datetime
+
+
+class Ticker:
+    def __init__(self, symbol: str):
+        self.symbol = symbol
+
+    def history(self, start=None, end=None, **kwargs):
+        class DummyData:
+            def __init__(self):
+                self.index = type("Idx", (), {"tz": None})()
+                self.columns = []
+                self._data = []
+
+            @property
+            def empty(self):
+                return True
+
+            def to_csv(self):
+                return ""
+
+            def __len__(self):
+                return 0
+
+        return DummyData()

yfinance/exceptions.py

@@ -0,0 +1,2 @@
+class YFRateLimitError(Exception):
+    pass