test app

.gitignore

@@ -1,46 +0,0 @@
-# Python virtual environments
-.venv/
-venv/
-ENV/
-env/
-.env
-
-# Python cache files
-__pycache__/
-*.py[cod]
-*$py.class
-*.so
-.Python
-build/
-develop-eggs/
-dist/
-downloads/
-eggs/
-.eggs/
-lib/
-lib64/
-parts/
-sdist/
-var/
-wheels/
-*.egg-info/
-.installed.cfg
-*.egg
-
-# IDE specific files
-.idea/
-.vscode/
-*.swp
-*.swo
-.DS_Store
-
-# Jupyter Notebook
-.ipynb_checkpoints
-
-# Environment variables
-.env
-.env.local
-.env.*.local
-
-# Logs
-*.log 

pyrightconfig.json

@@ -1,4 +0,0 @@
-{
-    "venvPath": ".",
-    "venv": ".venv"
-} 

requirements.txt

@@ -1,70 +1,3 @@
-altair==5.5.0
-annotated-types==0.7.0
-anyio==4.9.0
-async-timeout==4.0.3
-attrs==25.3.0
-blinker==1.9.0
-cachetools==5.5.2
-certifi==2025.4.26
-charset-normalizer==3.4.2
-click==8.2.1
-exceptiongroup==1.3.0
-filetype==1.2.0
-gitdb==4.0.12
-gitpython==3.1.44
-google-ai-generativelanguage==0.6.18
-google-api-core==2.25.1
-google-auth==2.40.3
-googleapis-common-protos==1.70.0
-grpcio==1.73.0
-grpcio-status==1.73.0
-h11==0.16.0
-httpcore==1.0.9
-httpx==0.28.1
-idna==3.10
-jinja2==3.1.6
-jsonpatch==1.33
-jsonpointer==3.0.0
-jsonschema==4.24.0
-jsonschema-specifications==2025.4.1
-langchain==0.3.25
-langchain-core==0.3.65
-langchain-google-genai==2.1.5
-langchain-text-splitters==0.3.8
-langsmith==0.3.45
-markupsafe==3.0.2
-narwhals==1.42.1
-numpy==2.2.6
-orjson==3.10.18
-packaging==24.2
-pandas==2.3.0
-pillow==11.2.1
-proto-plus==1.26.1
-protobuf==6.31.1
-pyarrow==20.0.0
-pyasn1==0.6.1
-pyasn1-modules==0.4.2
-pydantic==2.11.7
-pydantic-core==2.33.2
-pydeck==0.9.1
-python-dateutil==2.9.0.post0
-pytz==2025.2
-pyyaml==6.0.2
-referencing==0.36.2
-requests==2.32.4
-requests-toolbelt==1.0.0
-rpds-py==0.25.1
-rsa==4.9.1
-six==1.17.0
-smmap==5.0.2
-sniffio==1.3.1
-sqlalchemy==2.0.41
-streamlit==1.45.1
-tenacity==9.1.2
-toml==0.10.2
-tornado==6.5.1
-typing-extensions==4.14.0
-typing-inspection==0.4.1
-tzdata==2025.2
-urllib3==2.4.0
-zstandard==0.23.0
+altair
+pandas
+streamlit

src/llm.py

@@ -1,44 +0,0 @@
-import getpass
-import os
-from dotenv import load_dotenv
-from pydantic import BaseModel, Field
-from langchain.chat_models import init_chat_model
-
-class GraspTarget(BaseModel):
-    """Grasp target position for robot"""
-    # TODO: 需要根据实际的机器人坐标系进行调整， X coordinate in meters？
-    x: float = Field(description="x-axis coordinates in three-dimensional space")
-    y: float = Field(description="y-axis coordinates in three-dimensional space")
-    z: float = Field(description="z-axis coordinates in three-dimensional space")
-
-class GraspTargetGenerator:
-    """A class to generate grasp targets using LLM"""
-    
-    def __init__(self, model_name: str = "gemini-2.0-flash", model_provider: str = "google_genai"):
-        """Initialize the GraspTargetGenerator with specified model"""
-        # 加载环境变量
-        load_dotenv()
-        
-        # 确保有 API key
-        if not os.environ.get("GOOGLE_API_KEY"):
-            os.environ["GOOGLE_API_KEY"] = getpass.getpass("Enter API key for Google Gemini: ")
-        
-        # 初始化 LLM
-        self.llm = init_chat_model(model_name, model_provider=model_provider)
-        self.structured_llm = self.llm.with_structured_output(GraspTarget)
-    
-    def generate_grasp_target(self, task_description: str) -> list[float]:
-        """Generate grasp target coordinates based on task description
-        
-        Args:
-            task_description: Description of the task to generate coordinates for
-            
-        Returns:
-            List of [x, y, z] coordinates
-        """
-        response = self.structured_llm.invoke(f"Task: {task_description}")
-        return [response.x, response.y, response.z]
-
-if __name__ == "__main__":
-    generator = GraspTargetGenerator()
-    print(generator.generate_grasp_target("Pick up the red ball"))